| /external/tensorflow/tensorflow/compiler/aot/ |
| compile.h | 36 xla::ProgramShapeProto program_shape; // Static shape of args and results. member in struct:tensorflow::tfcompile::CompileResult
|
| codegen.cc | 396 const xla::ProgramShapeProto& ps = compile_result.program_shape;
502 // {{PROGRAM_SHAPE}}
666 {"{{PROGRAM_SHAPE}}", xla::ShapeUtil::HumanString(xla::ProgramShape(ps))},
694 std::unique_ptr<xla::ProgramShapeProto> program_shape; local
697 program_shape =
698 absl::make_unique<xla::ProgramShapeProto>(compile_result.program_shape);
703 program_shape->clear_parameter_names();
711 "xla::ProgramShapeProto", program_shape.get()};
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| computation_layout.cc | 26 ComputationLayout::ComputationLayout(const ProgramShape& program_shape,
28 : result_layout_(program_shape.result()) {
29 for (auto& shape : program_shape.parameters()) {
60 ProgramShape program_shape; local
62 *program_shape.add_parameters() = parameter_layouts_[i].shape();
63 *program_shape.add_parameter_names() = absl::StrCat("p", i);
65 *program_shape.mutable_result() = result_layout_.shape();
66 return program_shape;
|
| hlo_proto_util.cc | 62 const auto& program_shape = hlo_proto.hlo_module().host_program_shape(); local
63 for (const ShapeProto& shape : program_shape.parameters()) {
|
| hlo_computation.cc | 708 ProgramShape program_shape; local
711 *program_shape.add_parameters() = param_instruction->shape();
712 *program_shape.add_parameter_names() = param_instruction->name();
714 *program_shape.mutable_result() = root_instruction_->shape();
716 return program_shape;
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| testing.cc | 102 auto program_shape = computation.proto().host_program_shape(); local
105 for (const ShapeProto& shape : program_shape.parameters()) {
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| shape_test.cc | 143 ProgramShape program_shape; local
144 *program_shape.add_parameters() = ShapeUtil::MakeShape(F32, {1, 2, 3});
145 *program_shape.add_parameters() = ShapeUtil::MakeTokenShape();
146 *program_shape.add_parameters() = ShapeUtil::MakeShape(S64, {});
147 *program_shape.add_parameters() = ShapeUtil::MakeTupleShape(
152 *program_shape.mutable_result() = ShapeUtil::MakeShape(F32, {7});
154 program_shape.add_parameter_names("foo");
155 program_shape.add_parameter_names("bar");
156 program_shape.add_parameter_names("baz");
157 program_shape.add_parameter_names("qux qux")
[all...] |
| shape_util.cc | 150 ProgramShape program_shape; local
152 *program_shape.add_parameters() = shape;
154 *program_shape.mutable_result() = std::move(result);
155 return program_shape;
477 /* static */ string ShapeUtil::HumanString(const ProgramShape& program_shape) {
479 for (auto& shape : program_shape.parameters()) {
481 parameters.push_back(StrCat(i < program_shape.parameter_names_size()
482 ? program_shape.parameter_names(i)
487 HumanString(program_shape.result()));
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| dumped_computation_to_operation_list.cc | 77 std::unique_ptr<ProgramShape> program_shape = local
81 layouts.reserve(program_shape->parameters_size());
82 for (int i = 0; i < program_shape->parameters_size(); ++i) {
83 layouts.push_back(&program_shape->parameters(i));
87 build_options.set_result_layout(program_shape->result());
|
| dumped_computation_to_text.cc | 53 std::unique_ptr<ProgramShape> program_shape = local
57 layouts.reserve(program_shape->parameters_size());
58 for (int i = 0; i < program_shape->parameters_size(); ++i) {
59 layouts.push_back(&program_shape->parameters(i));
64 build_options.set_result_layout(program_shape->result());
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| xrt_compile_ops.cc | 115 config.program_shape().parameters_size());
117 config.program_shape().parameters_size());
118 for (int i = 0; i < config.program_shape().parameters_size(); ++i) {
119 argument_layouts[i] = xla::Shape(config.program_shape().parameters(i));
124 build_options.set_result_layout(xla::Shape(config.program_shape().result()));
187 xla::ProgramShapeProto program_shape = executable->executable() local
194 program_shape_output.vec<string>()(0) = program_shape.SerializeAsString();
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| xla_builder.cc | 159 ProgramShape program_shape; local
161 *program_shape.mutable_result() = Shape(root_proto->shape());
167 program_shape.add_parameters();
168 program_shape.add_parameter_names();
178 *program_shape.mutable_parameters(index) = Shape(instr.shape());
179 *program_shape.mutable_parameter_names(index) = instr.name();
182 return program_shape;
351 TF_ASSIGN_OR_RETURN(ProgramShape program_shape, GetProgramShape(root_id));
352 *entry.mutable_program_shape() = program_shape.ToProto();
368 *module->mutable_host_program_shape() = entry.program_shape();
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| xrt.cc | 154 ProgramShape program_shape; local
156 *program_shape.add_parameters() = shape;
158 *program_shape.mutable_result() = result_shape;
160 LayoutUtil::SetToDefaultLayout(&program_shape);
161 *config->mutable_program_shape() = program_shape.ToProto();
174 return new XrtExecutable(program_shape, handle, session_target);
177 XrtExecutable::XrtExecutable(const ProgramShape& program_shape, int64 handle,
179 : program_shape_(program_shape),
207 const int num_expected_arguments = program_shape().parameters().size();
241 return new XrtAllocation(output, program_shape().result(), session_target_)
244 const ProgramShape& XrtExecutable::program_shape() const { function in class:xla::swig::XrtExecutable
[all...] |
| /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;
238 const auto& program_shape = GetProgramShapeWithLayout(test_module); local
242 if (!ProgramShapesEqual(program_shape,
|
| /external/tensorflow/tensorflow/compiler/xrt/tests/ |
| raw_api_test.cc | 729 TF_EXPECT_OK(session.Run({read_back, c_handle.program_shape}, &outputs));
737 xla::ProgramShapeProto program_shape; local
738 EXPECT_TRUE(program_shape.ParseFromString(outputs[1].vec<string>()(0)));
739 EXPECT_EQ(program_shape.parameters_size(), 2);
784 TF_EXPECT_OK(session.Run({read_back, c_handle.program_shape}, &outputs));
792 xla::ProgramShapeProto program_shape; local
793 EXPECT_TRUE(program_shape.ParseFromString(outputs[1].vec<string>()(0)));
794 EXPECT_EQ(program_shape.parameters_size(), 2);
1231 xla::ProgramShapeProto program_shape; local
[all...] |
