| /external/tensorflow/tensorflow/cc/gradients/ |
| D | data_flow_grad_test.cc | 30 using ops::DynamicStitch;
66 auto y = DynamicStitch(scope_, indices, data); in TEST_F()
|
| D | data_flow_grad.cc | 99 DynamicStitch(scope, partitioned_indices.outputs, grad_inputs); in DynamicPartitionGrad()
|
| D | math_grad.cc | 672 return DynamicStitch(scope, indices, data); in ReducedShapeHelper()
|
| /external/tensorflow/tensorflow/core/api_def/java_api/ |
| D | api_def_DynamicStitch.pbtxt | 2 graph_op_name: "DynamicStitch"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_ParallelDynamicStitch.pbtxt | 30 DynamicStitch operator that defines the behavior in that case.
64 <img style="width:100%" src="https://www.tensorflow.org/images/DynamicStitch.png" alt>
|
| D | api_def_DynamicStitch.pbtxt | 2 graph_op_name: "DynamicStitch"
65 <img style="width:100%" src="https://www.tensorflow.org/images/DynamicStitch.png" alt>
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | DynamicStitch.pbtxt | 2 name: "DynamicStitch"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | DynamicStitch.pbtxt | 2 name: "DynamicStitch"
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | data_flow_ops_test.cc | 120 TEST(DataFlowOpsTest, DynamicStitch) { in TEST() argument
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | lower_tf.mlir | 611 %0 = "tf.DynamicStitch"(%indices, %arg0) : (tensor<2xi32>, tensor<2x2xf32>) -> tensor<2x2xf32>
632 …%0 = "tf.DynamicStitch"(%indices0, %indices1, %arg0, %arg1) : (tensor<i32>, tensor<2x2xi32>, tenso…
639 // CHECK-NOT: tf.DynamicStitch
642 %0 = "tf.DynamicStitch"(%indices, %arg0) : (tensor<2xi32>, tensor<2x2xui8>) -> tensor<2x2xui8>
656 %0 = "tf.DynamicStitch"(%indices, %arg0) : (tensor<2xi32>, tensor<2xf32>) -> tensor<2xf32>
670 %0 = "tf.DynamicStitch"(%indices, %arg0) : (tensor<2xi32>, tensor<2x2x2xf32>) -> tensor<2x2x2xf32>
676 // CHECK: tf.DynamicStitch
677 %0 = "tf.DynamicStitch"(%arg0, %arg1) : (tensor<*xi32>, tensor<*xf32>) -> tensor<*xf32>
690 %0 = "tf.DynamicStitch"(%indices, %arg0) : (tensor<2xi32>, tensor<2x2xf32>) -> tensor<1x2xf32>
|
| D | tf-ops.mlir | 3352 %0 = "tf.DynamicStitch"(%indices, %arg0) : (tensor<2xi32>, tensor<2x2xf32>) -> tensor<2x2xf32>
3360 %0 = "tf.DynamicStitch"() : () -> (tensor<2x2xf32>)
3369 %0 = "tf.DynamicStitch"(%indices, %arg0) : (tensor<2xi32>, tensor<2x2xf32>) -> tensor<f32>
3378 %0 = "tf.DynamicStitch"(%indices, %arg0) : (tensor<2xi32>, tensor<2x2xf32>) -> tensor<2x2xf32>
3387 %0 = "tf.DynamicStitch"(%indices, %arg0) : (tensor<2xi32>, tensor<3x2xf32>) -> tensor<2x2xf32>
3398 …%0 = "tf.DynamicStitch"(%indices0, %indices1, %arg0, %arg1) : (tensor<i32>, tensor<2x2xi32>, tenso…
3407 %0 = "tf.DynamicStitch"(%indices, %arg0) : (tensor<2xi32>, tensor<2x2xf32>) -> tensor<2x2xf32>
3416 %0 = "tf.DynamicStitch"(%indices, %arg0) : (tensor<2xi32>, tensor<2x2xf32>) -> tensor<3x2xf32>
3424 %0 = "tf.DynamicStitch"(%arg0, %arg1) : (tensor<?x2xi32>, tensor<?x3x3xf32>) -> tensor<*xf32>
3435 …%0 = "tf.DynamicStitch"(%indices0, %indices1, %arg0, %arg1) : (tensor<i32>, tensor<i32>, tensor<?x…
|
| D | constant-fold.mlir | 635 %0 = "tf.DynamicStitch"(%indices, %arg0) : (tensor<0xi32>, tensor<0x2xf32>) -> tensor<0x2xf32>
|
| /external/tensorflow/tensorflow/compiler/jit/tests/ |
| D | opens2s_gnmt_mixed_precision.golden_summary | 219 DynamicStitch 1
|
| D | opens2s_gnmt_mixed_precision.pbtxt.gz | |
| /external/tensorflow/tensorflow/lite/g3doc/guide/ |
| D | op_select_allowlist.md | 154 * `raw_ops.DynamicStitch`
|
| /external/tensorflow/tensorflow/compiler/tf2xla/g3doc/ |
| D | gpu_supported_ops.md | 70 `DynamicStitch` | `T={bool,complex64,double,float,int32,int64,uint32,uint64}`
|
| D | cpu_supported_ops.md | 70 `DynamicStitch` | `T={bool,complex64,double,float,int32,int64,uint32,uint64}`
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | tooling_util.cc | 425 HANDLE_OPERATORTYPENAME_CASE(DynamicStitch) in OperatorTypeName()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | function_test.cc | 1938 auto y_shape = ops::DynamicStitch( in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 1894 TEST_F(OpTest, DynamicStitch) { in TEST_F() argument
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/ |
| D | legalization.md | 1617 ### tf.DynamicStitch
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 3470 def TF_DynamicStitchOp : TF_Op<"DynamicStitch", [NoSideEffect, SameVariadicOperandSize]> {
3536 <img style="width:100%" src="https://www.tensorflow.org/images/DynamicStitch.png" alt>
9441 DynamicStitch operator that defines the behavior in that case.
9475 <img style="width:100%" src="https://www.tensorflow.org/images/DynamicStitch.png" alt>
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 1284 name: "DynamicStitch"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 1284 name: "DynamicStitch"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v0.pbtxt | 10860 name: "DynamicStitch"
10887 name: "DynamicStitch"
|