Home
last modified time | relevance | path

Searched refs:add_parameters (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dshape_test.cc144 *program_shape.add_parameters() = ShapeUtil::MakeShape(F32, {1, 2, 3}); in TEST_F()
145 *program_shape.add_parameters() = ShapeUtil::MakeTokenShape(); in TEST_F()
146 *program_shape.add_parameters() = ShapeUtil::MakeShape(S64, {}); in TEST_F()
147 *program_shape.add_parameters() = ShapeUtil::MakeTupleShape( in TEST_F()
Dshape.cc188 *add_parameters() = Shape(shape_proto); in ProgramShape()
199 *proto.add_parameters() = shape.ToProto(); in ToProto()
Dshape.h242 Shape* add_parameters() { in add_parameters() function
Dshape_util.cc152 *program_shape.add_parameters() = shape; in MakeProgramShape()
/external/tensorflow/tensorflow/compiler/xrt/tests/
Draw_api_test.cc450 *shapes1->add_parameters() = in TEST()
452 *shapes1->add_parameters() = in TEST()
461 *shapes2->add_parameters() = in TEST()
463 *shapes2->add_parameters() = in TEST()
698 *shapes->add_parameters() = in TEST()
700 *shapes->add_parameters() = in TEST()
751 *shapes->add_parameters() = in TEST()
753 *shapes->add_parameters() = in TEST()
821 *shapes->add_parameters() = param_shape.ToProto(); in TEST()
872 *shapes->add_parameters() = in TEST()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/
Dcomputation_layout.cc62 *program_shape.add_parameters() = parameter_layouts_[i].shape(); in ComputeProgramShape()
Dhlo_computation.cc711 *program_shape.add_parameters() = param_instruction->shape(); in ComputeProgramShape()
/external/tensorflow/tensorflow/compiler/xla/python/
Dxrt.cc156 *program_shape.add_parameters() = shape; in CompileForXrt()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dhlo_test_base.cc79 *program_shape.add_parameters() = param->shape(); in GetProgramShapeWithLayout()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc167 program_shape.add_parameters(); in GetProgramShape()