| /external/tensorflow/tensorflow/compiler/aot/ |
| compile.h | 36 xla::ProgramShape program_shape; // Static shape of args and results. member in struct:tensorflow::tfcompile::CompileResult
|
| codegen.cc | 103 const xla::ProgramShape& ps = compile_result.program_shape;
299 const xla::ProgramShape& ps = compile_result.program_shape;
389 // {{PROGRAM_SHAPE}}
420 data->program_shape = StaticProgramShape();
511 {"{{PROGRAM_SHAPE}}", xla::ShapeUtil::HumanString(ps)},
539 std::unique_ptr<xla::ProgramShape> program_shape; local
542 program_shape =
543 tensorflow::MakeUnique<xla::ProgramShape>(compile_result.program_shape);
547 program_shape->clear_parameter_names();
557 "xla::ProgramShape", program_shape.get()))
[all...] |
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| xla_jit_compiled_cpu_function_test.cc | 119 const xla::ProgramShape* program_shape = function.ProgramShape(); local
120 ASSERT_TRUE(program_shape != nullptr);
121 ASSERT_EQ(program_shape->parameters_size(), 2);
122 EXPECT_TRUE(ShapeUtil::Compatible(program_shape->parameters(0), s32));
123 EXPECT_TRUE(ShapeUtil::Compatible(program_shape->parameters(1), s32));
125 const xla::Shape& result = program_shape->result();
|
| xla_compiled_cpu_function.h | 78 const xla::ProgramShape* program_shape = nullptr; member in struct:tensorflow::XlaCompiledCpuFunction::StaticData
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| testing.cc | 101 auto program_shape = local
107 for (const Shape& parameter : program_shape->parameters()) {
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| dumped_computation_to_operation_list.cc | 80 std::unique_ptr<ProgramShape> program_shape = local
84 layouts.reserve(program_shape->parameters_size());
85 for (int i = 0; i < program_shape->parameters_size(); ++i) {
86 layouts.push_back(&program_shape->parameters(i));
90 build_options.set_result_layout(program_shape->result());
|
| dumped_computation_to_text.cc | 55 std::unique_ptr<ProgramShape> program_shape = local
59 layouts.reserve(program_shape->parameters_size());
60 for (int i = 0; i < program_shape->parameters_size(); ++i) {
61 layouts.push_back(&program_shape->parameters(i));
66 build_options.set_result_layout(program_shape->result());
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| hlo_test_base.cc | 76 ProgramShape program_shape; local
79 *program_shape.add_parameters() = param->shape();
80 *program_shape.add_parameter_names() = param->name();
82 *program_shape.mutable_result() = entry->root_instruction()->shape();
83 return program_shape;
128 const auto& program_shape = GetProgramShapeWithLayout(test_module); local
132 if (!ProgramShapesEqual(program_shape,
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_module.cc | 260 ProgramShape program_shape; local
261 *program_shape.mutable_result() = root->shape();
267 *program_shape.add_parameters() = shape;
268 program_shape.add_parameter_names(name);
271 return std::move(program_shape);
357 TF_ASSIGN_OR_RETURN(ProgramShape program_shape,
360 HloModuleConfig module_config(program_shape);
369 program_shape.parameters(i)));
372 program_shape.result()));
|
| hlo_computation.cc | 527 ProgramShape program_shape; local
530 *program_shape.add_parameters() = param_instruction->shape();
531 *program_shape.add_parameter_names() = param_instruction->name();
533 *program_shape.mutable_result() = root_instruction_->shape();
535 LayoutUtil::ClearLayout(&program_shape);
536 return program_shape;
|
| service.cc | 265 const ProgramShape& program_shape,
269 auto config = MakeUnique<HloModuleConfig>(program_shape);
272 if (program_shape.parameters_size() != argument_shapes.size()) {
274 program_shape.parameters_size(),
281 program_shape.parameters(i))) {
286 i, ShapeUtil::HumanString(program_shape.parameters(i)).c_str(),
298 program_shape.result()));
325 const ProgramShape& program_shape,
333 return CreateModuleConfig(program_shape, argument_shapes, &execution_options,
742 std::shared_ptr<const ProgramShape> program_shape,
1235 ProgramShape program_shape; local
[all...] |
| user_computation.cc | 1621 auto program_shape = MakeUnique<ProgramShape>(); local
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/ |
| shape_util.cc | 174 ProgramShape program_shape; local
176 *program_shape.add_parameters() = shape;
178 *program_shape.mutable_result() = std::move(result);
179 return program_shape;
484 /* static */ string ShapeUtil::HumanString(const ProgramShape& program_shape) {
486 for (auto& shape : program_shape.parameters()) {
489 tensorflow::strings::StrCat(i < program_shape.parameter_names_size()
490 ? program_shape.parameter_names(i)
496 HumanString(program_shape.result()));
[all...] |
