Home
last modified time | relevance | path

Searched defs:reshape_op (Results 1 – 24 of 24) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dconvert_reorder_axes.cc31 auto* reshape_op = new TensorFlowReshapeOperator; in CreateReshapeFromReorderAxes() local
134 auto* reshape_op = in Run() local
Dresolve_reshape_attributes.cc33 auto* reshape_op = reshape_it->get(); in Run() local
Dconvert_squeeze_to_reshape.cc68 auto* reshape_op = new TensorFlowReshapeOperator; in Run() local
Dremove_trivial_reshape.cc89 auto* reshape_op = reshape_it->get(); in Run() local
Dconvert_trivial_pack_to_reshape.cc57 auto* reshape_op = new TensorFlowReshapeOperator; in Run() local
Dconvert_expanddims_to_reshape.cc72 auto* reshape_op = new TensorFlowReshapeOperator; in Run() local
Dmove_binary_operator_before_reshape.cc121 Operator* reshape_op = reshape_it->get(); in Run() local
Dconvert_trivial_transpose_to_reshape.cc86 auto* reshape_op = new TensorFlowReshapeOperator; in Run() local
Dmerge_reshape_into_preceding_transpose.cc109 auto* reshape_op = ConvertOperator<TensorFlowReshapeOperator*>( in Run() local
Dreorder_reshape_transpose.cc128 TensorFlowReshapeOperator* reshape_op = in Run() local
Didentify_nearest_upsample.cc187 auto* reshape_op = new TensorFlowReshapeOperator; in Run() local
Dunroll_batch_matmul.cc46 auto* reshape_op = new TensorFlowReshapeOperator; in SliceInput() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dunroll_batch_matmul.cc89 auto reshape_op = createReshapeOp(value, {batch_size, num_rows, num_cols}, in sliceInput() local
303 auto reshape_op = createReshapeOp(pack_op.output(), result_shape, in matchAndRewrite() local
Doptimize.cc56 auto reshape_op = llvm::dyn_cast_or_null<ReshapeOp>(user); in matchAndRewrite() local
Dtpu_space_to_depth_pass.cc296 auto reshape_op = in HandleConv2DFilter() local
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
Drsqrt_builder.cc112 auto* reshape_op = graph_builder_->AddNode(GetTFLiteNodeID()); in PopulateSubGraph() local
/external/tensorflow/tensorflow/c/
Dc_api_experimental_test.cc205 TFE_Op* reshape_op = TFE_NewOp(tfe_context_, "Reshape", status_); in TEST_F() local
/external/tensorflow/tensorflow/lite/toco/
Dexport_tensorflow.cc613 tensorflow::NodeDef* reshape_op = tensorflow_graph->add_node(); in ConvertFullyConnectedOperator() local
828 tensorflow::NodeDef* reshape_op = tensorflow_graph->add_node(); in ConvertSoftmaxOperator() local
870 tensorflow::NodeDef* reshape_op = tensorflow_graph->add_node(); in ConvertLogSoftmaxOperator() local
1052 tensorflow::NodeDef* reshape_op = tensorflow_graph->add_node(); in ConvertTensorFlowReshapeOperator() local
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_common.cc397 auto reshape_op = rewriter.create<tosa::ReshapeOp>( in convertSelectOp() local
2360 auto reshape_op = rewriter.create<tosa::ReshapeOp>( in convertReduceOpCommon() local
Dlegalize_tfl.cc1478 auto reshape_op = rewriter.create<tosa::ReshapeOp>( in matchAndRewrite() local
/external/tensorflow/tensorflow/core/kernels/
Deigen_spatial_convolutions_test.cc1515 const auto reshape_op = in PackRhsHelper() local
1688 const auto reshape_op = in PackLhsHelper() local
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlegalize_to_linalg.cc721 OpTy reshape_op, ArrayRef<Value> args, in matchAndRewrite()
/external/tensorflow/tensorflow/tools/graph_transforms/
Dquantize_nodes_test.cc479 Output reshape_op = in TestQuantizeReshape() local
/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator_test.cc2073 for (auto reshape_op : reshape_ops) { in TEST_F() local