Home
last modified time | relevance | path

Searched refs:add_parameter_names (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dshape_test.cc154 program_shape.add_parameter_names("foo"); in TEST_F()
155 program_shape.add_parameter_names("bar"); in TEST_F()
156 program_shape.add_parameter_names("baz"); in TEST_F()
157 program_shape.add_parameter_names("qux qux"); in TEST_F()
194 prog.add_parameter_names("arg0"); in TEST_F()
195 prog.add_parameter_names("scalar"); in TEST_F()
196 prog.add_parameter_names("matrix"); in TEST_F()
197 prog.add_parameter_names("matrix2"); in TEST_F()
198 prog.add_parameter_names("tuple"); in TEST_F()
199 prog.add_parameter_names("nested_tuple"); in TEST_F()
Dshape.cc192 add_parameter_names(name); in ProgramShape()
203 proto.add_parameter_names(name); in ToProto()
Dshape.h265 void add_parameter_names(const string& value) { in add_parameter_names() function
268 string* add_parameter_names() { in add_parameter_names() function
/external/tensorflow/tensorflow/compiler/xla/service/
Dcomputation_layout.cc63 *program_shape.add_parameter_names() = absl::StrCat("p", i); in ComputeProgramShape()
Dhlo_computation.cc712 *program_shape.add_parameter_names() = param_instruction->name(); in ComputeProgramShape()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dhlo_test_base.cc80 *program_shape.add_parameter_names() = param->name(); in GetProgramShapeWithLayout()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc168 program_shape.add_parameter_names(); in GetProgramShape()