Searched refs:reshape_shape (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_instruction_fusion_test.cc | 337 Shape reshape_shape = ShapeUtil::MakeShape(F32, {8, 8}); in TEST_F() local
348 HloInstruction::CreateReshape(reshape_shape, broadcast2)); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier_test.cc | 1983 Shape reshape_shape = builder in TEST_F() local
1997 reshape_shape)); in TEST_F()
3552 Shape reshape_shape = ShapeUtil::MakeShape(F32, {30, 1, 4}); in TEST_F() local
3554 HloInstruction::CreateReshape(reshape_shape, transpose)); in TEST_F()
3561 EXPECT_TRUE(ShapeUtil::Equal(root->shape(), reshape_shape)); in TEST_F()
3567 .WithShapeEqualTo(&reshape_shape))); in TEST_F()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | export_tensorflow.cc | 601 const string reshape_shape = in ConvertFullyConnectedOperator() local
606 CreateMatrixShapeTensorConst(reshape_shape, fc_weights_shape.dims(1), -1, in ConvertFullyConnectedOperator()
612 reshape_op->add_input(reshape_shape); in ConvertFullyConnectedOperator()
|