| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | replicate_invariant_op_hoisting.cc | 47 Block* replicate_block, TF::ShapeOp shape_op) { in MakeShapeOpInvariant()
128 replicate_op.walk([&](TF::ShapeOp shape_op) { in HoistReplicateInvariantOps()
|
| D | tpu_extract_outside_compilation.cc | 531 func::FuncOp shape_op = BuildFunction( in MoveOpsToHost() local
|
| D | tpu_rewrite_pass.cc | 356 auto shape_op = builder->create<TF::ShapeOp>( in BuildCompileOp() local
|
| D | tensor_list_ops_decomposition.cc | 527 auto shape_op = shape_value.getDefiningOp(); in GetConstShapeValue() local
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | propagate_array_data_types.cc | 81 auto* shape_op = static_cast<TensorFlowShapeOperator*>(op); in Run() local
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | unary_ops_test.py | 999 shape_op = lambda x: array_ops.shape_internal(x, optimize=False) function
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | fold_old_batch_norms_test.cc | 395 Output shape_op = in TestFoldFusedBatchNormsWithConcat() local
|
| D | quantize_nodes_test.cc | 373 Output shape_op = in TestQuantizeConcat() local
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_experimental_test.cc | 458 TFE_Op* shape_op = ShapeOp(ctx, hgpu); in TEST() local
|
| D | c_api_test.cc | 571 TFE_Op* shape_op = ShapeOp(ctx, hgpu); in TEST() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | flatbuffer_import.cc | 843 auto shape_op = builder.create<tfl::ConstOp>(loc, output_shape); in ConvertOp() local
905 auto shape_op = builder.create<tfl::ConstOp>(loc, output_shape); in ConvertOp() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/ir/ |
| D | tfr_ops.cc | 500 LogicalResult matchAndRewrite(GetShapeOp shape_op, in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 256 auto shape_op = dyn_cast_or_null<ShapeOp>(slice_op.input().getDefiningOp()); in fold() local
2048 auto shape_op = input().getDefiningOp<ShapeOp>(); in fold() local
|
| D | tf_ops_a_m.cc | 2648 if (auto shape_op = dims.getDefiningOp<ShapeOp>()) { in InferFillOpType() local
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | export_tensorflow.cc | 1315 tensorflow::NodeDef* shape_op = tensorflow_graph->add_node(); in ConvertTensorFlowShapeOperator() local
|
| /external/tensorflow/tensorflow/core/transforms/constant_folding/ |
| D | pass.cc | 2120 Operation *shape_op = op->getOperand(1).getDefiningOp(); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 6390 auto shape_op = in matchAndRewrite() local
|