| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_module_config.cc | 31 HloModuleConfig::HloModuleConfig(const ProgramShape& program_shape, in HloModuleConfig()
37 const ProgramShape& program_shape) { in SetDefaultComputationLayout()
|
| D | computation_layout.cc | 26 ComputationLayout::ComputationLayout(const ProgramShape& program_shape, in ComputationLayout()
60 ProgramShape program_shape; in ComputeProgramShape() local
|
| D | local_service.cc | 99 const ProgramShape* program_shape) { in CreateExecutionOptions()
125 ProgramShape program_shape(proto.host_program_shape()); in CompileExecutable() local
|
| D | hlo_proto_util.cc | 62 const auto& program_shape = hlo_proto.hlo_module().host_program_shape(); in EntryComputationParameterShapes() local
|
| D | service.cc | 265 const ProgramShape& program_shape, in CreateModuleConfig()
336 const ProgramShape& program_shape, in CreateModuleConfig()
1086 ProgramShape program_shape(arg->computation().host_program_shape()); in ComputeConstantGraph() local
|
| D | hlo_module.cc | 373 ProgramShape program_shape(module.host_program_shape()); in CreateModuleConfigFromProto() local
|
| D | hlo.proto | 227 xla.ProgramShapeProto program_shape = 4; field
|
| D | hlo_computation.cc | 708 ProgramShape program_shape; in ComputeProgramShape() local
|
| /external/tensorflow/tensorflow/compiler/aot/ |
| D | compile.h | 36 xla::ProgramShapeProto program_shape; // Static shape of args and results. member
|
| D | codegen.cc | 694 std::unique_ptr<xla::ProgramShapeProto> program_shape; in GenerateMetadata() local
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xrt.cc | 154 ProgramShape program_shape; in CompileForXrt() local
177 XrtExecutable::XrtExecutable(const ProgramShape& program_shape, int64 handle, in XrtExecutable()
244 const ProgramShape& XrtExecutable::program_shape() const { in program_shape() function in xla::swig::XrtExecutable
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | layout_util.cc | 166 /* static */ void LayoutUtil::SetToDefaultLayout(ProgramShape* program_shape) { in SetToDefaultLayout()
266 /* static */ void LayoutUtil::ClearLayout(ProgramShape* program_shape) { in ClearLayout()
318 /* static */ bool LayoutUtil::HasLayout(const ProgramShape& program_shape) { in HasLayout()
|
| D | shape_test.cc | 143 ProgramShape program_shape; in TEST_F() local
|
| D | shape.cc | 219 std::ostream& operator<<(std::ostream& out, const ProgramShape& program_shape) { in operator <<()
|
| D | shape_util.cc | 150 ProgramShape program_shape; in MakeProgramShape() local
477 /* static */ string ShapeUtil::HumanString(const ProgramShape& program_shape) { in HumanString()
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | dumped_computation_to_text.cc | 53 std::unique_ptr<ProgramShape> program_shape = in RealMain() local
|
| D | dumped_computation_to_operation_list.cc | 77 std::unique_ptr<ProgramShape> program_shape = in RealMain() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | testing.cc | 102 auto program_shape = computation.proto().host_program_shape(); in MakeFakeArgumentsOrDie() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_jit_compiled_cpu_function_test.cc | 120 const xla::ProgramShape program_shape(*function.ProgramShape()); in TEST() local
|
| D | xla_compiled_cpu_function.h | 297 StaticData* static_data, const xla::ProgramShapeProto* program_shape) { in set_static_data_program_shape()
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt.proto | 40 xla.ProgramShapeProto program_shape = 4; field
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_compile_ops.cc | 187 xla::ProgramShapeProto program_shape = executable->executable() in Compute() local
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | hlo_test_base.cc | 76 ProgramShape program_shape; in GetProgramShapeWithLayout() local
238 const auto& program_shape = GetProgramShapeWithLayout(test_module); in MakeReferenceModule() local
|
| /external/tensorflow/tensorflow/compiler/xrt/tests/ |
| D | raw_api_test.cc | 737 xla::ProgramShapeProto program_shape; in TEST() local
792 xla::ProgramShapeProto program_shape; in TEST() local
839 xla::ProgramShape program_shape(program_shape_proto); in TEST() local
1231 xla::ProgramShapeProto program_shape; in TEST() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 159 ProgramShape program_shape; in GetProgramShape() local
390 HloModuleProto* module, const ProgramShape& program_shape, in PopulateInputOutputAlias()
2545 ProgramShapeProto* program_shape = entry.mutable_program_shape(); in BuildConstantSubGraph() local
|