| /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_rewrite_pass.cc | 355 auto shape_op = builder->create<TF::ShapeOp>( in BuildCompileOp() local
|
| D | tpu_extract_outside_compilation.cc | 532 FuncOp shape_op = BuildFunction(clustered_ops.getArrayRef(), in MoveOpsToHost() local
|
| D | tensor_list_ops_decomposition.cc | 521 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 | 1007 shape_op = lambda x: array_ops.shape_internal(x, optimize=False) function
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_experimental_test.cc | 454 TFE_Op* shape_op = ShapeOp(ctx, hgpu); in TEST() local
|
| D | c_api_test.cc | 570 TFE_Op* shape_op = ShapeOp(ctx, hgpu); in TEST() local
|
| /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/compiler/mlir/lite/ |
| D | flatbuffer_import.cc | 814 auto shape_op = builder.create<tfl::ConstOp>(loc, output_shape); in ConvertOp() local
875 auto shape_op = builder.create<tfl::ConstOp>(loc, output_shape); in ConvertOp() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/ir/ |
| D | tfr_ops.cc | 468 LogicalResult matchAndRewrite(GetShapeOp shape_op, in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 239 auto shape_op = dyn_cast_or_null<ShapeOp>(slice_op.input().getDefiningOp()); in fold() local
1963 auto shape_op = input().getDefiningOp<ShapeOp>(); in fold() local
|
| D | tf_ops_a_m.cc | 2479 if (auto shape_op = dims.getDefiningOp<ShapeOp>()) { in InferFillOpType() local
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | export_tensorflow.cc | 1310 tensorflow::NodeDef* shape_op = tensorflow_graph->add_node(); in ConvertTensorFlowShapeOperator() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 5904 Value shape_op = rewriter.create<shape::ShapeOfOp>(op.getLoc(), x); in matchAndRewrite() local
6625 auto shape_op = in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | hlo_ops.cc | 645 Value shape_op) { in castToIndexTensor()
|