| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | unroll_batch_matmul.cc | 110 auto reshape_op = createReshapeOp(value, {num_rows, num_cols}, element_type, in sliceInput() local
115 auto reshape_op = createReshapeOp(value, {batch_size, num_rows, num_cols}, in sliceInput() local
135 auto reshape_op = createReshapeOp(split_value, {num_rows, num_cols}, in sliceInput() local
280 auto reshape_op = createReshapeOp(pack_op.output(), result_shape, in matchAndRewrite() local
|
| D | optimize.cc | 58 auto reshape_op = llvm::dyn_cast_or_null<ReshapeOp>(user); in matchAndRewrite() local
|
| D | tpu_space_to_depth_pass.cc | 251 auto reshape_op = in HandleConv2DFilter() local
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | convert_reorder_axes.cc | 31 auto* reshape_op = new TensorFlowReshapeOperator; in CreateReshapeFromReorderAxes() local
134 auto* reshape_op = in Run() local
|
| D | resolve_reshape_attributes.cc | 33 auto* reshape_op = reshape_it->get(); in Run() local
|
| D | convert_squeeze_to_reshape.cc | 68 auto* reshape_op = new TensorFlowReshapeOperator; in Run() local
|
| D | remove_trivial_reshape.cc | 89 auto* reshape_op = reshape_it->get(); in Run() local
|
| D | convert_trivial_pack_to_reshape.cc | 57 auto* reshape_op = new TensorFlowReshapeOperator; in Run() local
|
| D | convert_expanddims_to_reshape.cc | 72 auto* reshape_op = new TensorFlowReshapeOperator; in Run() local
|
| D | move_binary_operator_before_reshape.cc | 122 Operator* reshape_op = reshape_it->get(); in Run() local
|
| D | convert_trivial_transpose_to_reshape.cc | 87 auto* reshape_op = new TensorFlowReshapeOperator; in Run() local
|
| D | merge_reshape_into_preceding_transpose.cc | 109 auto* reshape_op = ConvertOperator<TensorFlowReshapeOperator*>( in Run() local
|
| D | reorder_reshape_transpose.cc | 128 TensorFlowReshapeOperator* reshape_op = in Run() local
|
| D | identify_nearest_upsample.cc | 188 auto* reshape_op = new TensorFlowReshapeOperator; in Run() local
|
| D | unroll_batch_matmul.cc | 47 auto* reshape_op = new TensorFlowReshapeOperator; in SliceInput() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/ |
| D | device_transform_patterns.cc | 252 auto reshape_op = in matchAndRewrite() local
408 auto reshape_op = in matchAndRewrite() local
558 auto reshape_op = in matchAndRewrite() local
652 auto reshape_op = in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/transforms/ |
| D | tf_jitrt_fusion.cc | 94 static bool IsUnitDimExpansionOnly(TensorReshapeOp reshape_op) { in IsUnitDimExpansionOnly()
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| D | rsqrt_builder.cc | 112 auto* reshape_op = graph_builder_->AddNode(GetTFLiteNodeID()); in PopulateSubGraph() local
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_experimental_test.cc | 205 TFE_Op* reshape_op = TFE_NewOp(tfe_context_, "Reshape", status_); in TEST_F() local
|
| /external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| D | meta_spmd_expander.cc | 774 auto reshape_op = mlir::cast<mlir::TF::ReshapeOp>(op); in ExpandOp() local
807 auto reshape_op = mlir::cast<mlir::TF::ReshapeOp>(op); in ComputeLayoutForward() local
840 auto reshape_op = mlir::cast<mlir::TF::ReshapeOp>(op); in ComputeLayoutBackward() local
|
| /external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/utils/ |
| D | tf_to_xla_attribute_utils.cc | 114 auto reshape_op = [&](Value value, const SmallVector<int64_t> &shape) { in PadForDynamicShapedInputSamePadding() local
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | export_tensorflow.cc | 618 tensorflow::NodeDef* reshape_op = tensorflow_graph->add_node(); in ConvertFullyConnectedOperator() local
833 tensorflow::NodeDef* reshape_op = tensorflow_graph->add_node(); in ConvertSoftmaxOperator() local
875 tensorflow::NodeDef* reshape_op = tensorflow_graph->add_node(); in ConvertLogSoftmaxOperator() local
1057 tensorflow::NodeDef* reshape_op = tensorflow_graph->add_node(); in ConvertTensorFlowReshapeOperator() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_common.cc | 437 auto reshape_op = CreateOpAndInfer<tosa::ReshapeOp>( in convertSelectOp() local
2641 auto reshape_op = CreateOpAndInfer<tosa::ReshapeOp>( in convertGenericReduceOp() local
2721 auto reshape_op = CreateOpAndInfer<tosa::ReshapeOp>( in convertReduceOpCommon() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_spatial_convolutions_test.cc | 1515 const auto reshape_op = in PackRhsHelper() local
1688 const auto reshape_op = in PackLhsHelper() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | optimize.cc | 1375 auto reshape_op = input.getDefiningOp<TFL::ReshapeOp>(); in matchAndRewrite() local
|