Searched refs:reshape_shape (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | optimize.cc | 62 ArrayRef<int64_t> reshape_shape = reshape_type.getShape(); in matchAndRewrite() local
72 if (reshape_shape.size() > input_shape.size()) return failure(); in matchAndRewrite()
95 if (reshape_shape.size() < non_unit_dims.size()) return failure(); in matchAndRewrite()
100 for (int64_t dim : reshape_shape) { in matchAndRewrite()
124 GetI64ConstantTensor(rewriter, reshape_shape, op.getLoc()); in matchAndRewrite()
|
| D | einsum.cc | 330 std::vector<int64_t> reshape_shape = in rewriteToBatchMatmul() local
332 if (failed(VerifyShapeOfReshapeOp(reshape_shape))) return failure(); in rewriteToBatchMatmul()
340 out = createReshapeOp(out, reshape_shape, original_type.getElementType(), in rewriteToBatchMatmul()
|
| D | legalize_hlo.cc | 743 std::vector<int64_t> reshape_shape(type.getRank(), 1); in matchAndRewrite() local
744 reshape_shape[iota_op.iota_dimension()] = type.getShape()[dimension]; in matchAndRewrite()
745 auto reshape_type = RankedTensorType::get(reshape_shape, element_type); in matchAndRewrite()
747 iota_op.getLoc(), rewriter.getI64TensorAttr(reshape_shape)); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| D | rsqrt_builder.cc | 108 const int reshape_shape[] = {1, 1, 1, 4}; in PopulateSubGraph() local
110 reshape_shape, reinterpret_cast<char*>(flat_shape), in PopulateSubGraph()
155 reshape_shape, reinterpret_cast<char*>(output_shape), in PopulateSubGraph()
|
| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | reshape_tester.cc | 127 const std::array<int32_t, 1> reshape_shape{ in CreateTfLiteModel() local
132 reshape_shape.data(), reshape_shape.size()), in CreateTfLiteModel()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_instruction_fusion_test.cc | 339 Shape reshape_shape = ShapeUtil::MakeShape(F32, {8, 8}); in TEST_F() local
350 HloInstruction::CreateReshape(reshape_shape, broadcast2)); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | hlo_ops.cc | 244 llvm::SmallVector<int64_t, 8> reshape_shape; in matchAndRewrite() local
247 reshape_shape.push_back(slice_shape[i]); in matchAndRewrite()
250 auto reshape_type = RankedTensorType::get(reshape_shape, element_type); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | export_tensorflow.cc | 606 const std::string reshape_shape = in ConvertFullyConnectedOperator() local
611 CreateMatrixShapeTensorConst(reshape_shape, fc_weights_shape.dims(1), -1, in ConvertFullyConnectedOperator()
617 reshape_op->add_input(reshape_shape); in ConvertFullyConnectedOperator()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier_test.cc | 2521 Shape reshape_shape = builder in TEST_F() local
2535 reshape_shape)); in TEST_F()
4320 Shape reshape_shape = ShapeUtil::MakeShape(F32, {30, 1, 4}); in TEST_F() local
4322 HloInstruction::CreateReshape(reshape_shape, transpose)); in TEST_F()
4329 EXPECT_TRUE(ShapeUtil::Equal(root->shape(), reshape_shape)); in TEST_F()
4335 .WithShapeEqualTo(&reshape_shape))); in TEST_F()
|