Searched refs:data_param (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dynamic_dimension_inference_test.cc | 159 auto data_param = builder.AddInstruction( in TEST_F() local
165 HloInstruction::CreateUnary(input_shape, HloOpcode::kNegate, data_param)); in TEST_F()
183 auto data_param = builder.AddInstruction( in TEST_F() local
189 HloInstruction::CreateUnary(input_shape, HloOpcode::kNegate, data_param)); in TEST_F()
215 auto data_param = builder.AddInstruction( in TEST_F() local
221 HloInstruction::CreateUnary(input_shape, HloOpcode::kNegate, data_param)); in TEST_F()
657 auto data_param = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F() local
663 ShapeUtil::MakeShape(F32, {5, 7}), data_param, /*start_indices=*/{0, 0}, in TEST_F()
|
| D | dynamic_padder_test.cc | 76 auto data_param = builder.AddInstruction( in TEST_F() local
82 HloInstruction::CreateUnary(input_shape, HloOpcode::kNegate, data_param)); in TEST_F()
|
| D | copy_insertion_test.cc | 895 auto data_param = builder.AddInstruction( in TEST_F() local
898 HloInstruction::CreateTuple({iter_param, data_param})); in TEST_F()
952 auto data_param = builder.AddInstruction( in TEST_F() local
958 data_param->shape(), HloOpcode::kExp, data_param)); in TEST_F()
1254 auto data_param = builder.AddInstruction( in TEST_F() local
1259 HloInstruction::CreateTuple({iter_param, data_param, data_param})); in TEST_F()
|