| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_module_config.cc | 31 HloModuleConfig::HloModuleConfig(const ProgramShape& program_shape, in HloModuleConfig()
40 const ProgramShape& program_shape) { in SetDefaultComputationLayout()
45 const ProgramShape& program_shape) { in SetComputationLayoutIfExists()
|
| H A D | computation_layout.cc | 27 ComputationLayout::ComputationLayout(const ProgramShape& program_shape, in ComputationLayout()
75 ProgramShape program_shape; in ComputeProgramShape() local
|
| H A D | service.cc | 261 const ProgramShape& program_shape, in CreateModuleConfig()
278 const ProgramShape& program_shape, in CreateModuleConfig()
1131 ProgramShape program_shape(arg->computation().host_program_shape()); in ComputeConstantGraph() local
|
| H A D | hlo_proto_util.cc | 66 const auto& program_shape = hlo_proto.hlo_module().host_program_shape(); in EntryComputationParameterShapes() local
|
| H A D | hlo_module.cc | 525 const ProgramShape& program_shape, const DebugOptions& debug_options, in CreateModuleConfigFromShape()
591 ProgramShape program_shape(module.host_program_shape()); in CreateModuleConfigFromProto() local
|
| H A D | hlo_module_util.cc | 45 const ProgramShape& program_shape, in CreateModuleConfig()
|
| H A D | local_service.cc | 105 ProgramShape program_shape(proto.host_program_shape()); in GetHloModuleConfig() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/ |
| H A D | layout_util.cc | 168 /* static */ void LayoutUtil::SetToDefaultLayout(ProgramShape* program_shape) { in SetToDefaultLayout()
278 /* static */ void LayoutUtil::ClearLayout(ProgramShape* program_shape) { in ClearLayout()
371 /* static */ bool LayoutUtil::HasLayout(const ProgramShape& program_shape) { in HasLayout()
|
| H A D | shape.cc | 255 std::ostream& operator<<(std::ostream& out, const ProgramShape& program_shape) { in operator <<()
|
| H A D | shape_test.cc | 146 ProgramShape program_shape; in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | xla_jit_compiled_cpu_function_test.cc | 220 const xla::ProgramShape program_shape(*function.ProgramShape()); in TEST() local
271 const xla::ProgramShape program_shape(*function.ProgramShape()); in TEST() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/ |
| H A D | tpu_compile_op_support.cc | 59 const ProgramShape& program_shape, absl::Span<const Shape> argument_shapes, in CreateModuleConfig()
137 const xla::ProgramShape& program_shape, in CreateModuleConfig()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/ |
| H A D | pod_tpu_driver.cc | 800 xla::ProgramShapeProto* program_shape) in GetCompiledProgramShape()
966 xla::Status PodCompiledProgramHandle::program_shape( in program_shape() function in tpu_driver::__anonf243d9070111::PodCompiledProgramHandle
967 xla::ProgramShapeProto* program_shape) { in program_shape()
|
| H A D | direct_tpu_driver.cc | 156 xla::Status program_shape(xla::ProgramShapeProto* program_shape) override { in program_shape() argument
|
| H A D | recording_tpu_driver.cc | 98 xla::Status program_shape(xla::ProgramShapeProto* program_shape) override { in program_shape() argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/aot/ |
| H A D | compile.h | 36 xla::ProgramShapeProto program_shape; // Static shape of args and results. member
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/ |
| H A D | compile_only_client.cc | 28 const ProgramShape& program_shape, in CreateModuleConfig()
|
| H A D | executable_build_options.cc | 181 const ProgramShape* program_shape) { in CreateExecutionOptions()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tools/ |
| H A D | dumped_computation_to_text.cc | 55 std::unique_ptr<ProgramShape> program_shape = in RealMain() local
|
| H A D | dumped_computation_to_operation_list.cc | 79 std::unique_ptr<ProgramShape> program_shape = in RealMain() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| H A D | testing.cc | 103 auto program_shape = computation.proto().host_program_shape(); in MakeFakeArgumentsOrDie() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | hlo_test_base.cc | 64 ProgramShape program_shape; in GetProgramShapeWithLayout() local
295 const auto& program_shape = GetProgramShapeWithLayout(test_module); in MakeReferenceModule() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/ |
| H A D | tpu_embedding_optimization_parameters_utils.cc | 168 const xla::ProgramShapeProto& program_shape = in GetBaseAuxiliaryParameterCount() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| H A D | xrt_compile_ops.cc | 230 xla::ProgramShapeProto program_shape = executable->executable() in Compute() local
|
| H A D | tpu_compile_ops.cc | 215 xla::ProgramShapeProto program_shape = in Compute() local
|