Home
last modified time | relevance | path

Searched refs:BroadcastGradientArgs (Results 1 – 25 of 28) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dconstant-fold.mlir506 …%r0, %r1 = "tf.BroadcastGradientArgs"(%s0, %s1) {} : (tensor<2xi32>, tensor<2xi32>) -> (tensor<0xi…
509 // CHECK-NOT: tf.BroadcastGradientArgs
519 …%r0, %r1 = "tf.BroadcastGradientArgs"(%s0, %s1) {} : (tensor<1xi32>, tensor<2xi32>) -> (tensor<1xi…
522 // CHECK-NOT: tf.BroadcastGradientArgs
532 …%r2, %r3 = "tf.BroadcastGradientArgs"(%s2, %s3) {} : (tensor<4xi32>, tensor<4xi32>) -> (tensor<1xi…
535 // CHECK-NOT: tf.BroadcastGradientArgs
545 …%r4, %r5 = "tf.BroadcastGradientArgs"(%s4, %s5) {} : (tensor<0xi32>, tensor<3xi32>) -> (tensor<3xi…
547 // CHECK-NOT: tf.BroadcastGradientArgs
557 …%r4, %r5 = "tf.BroadcastGradientArgs"(%s4, %s5) {} : (tensor<3xi32>, tensor<0xi32>) -> (tensor<2xi…
560 // CHECK-NOT: tf.BroadcastGradientArgs
[all …]
Dmark_ops_for_outside_compilation.mlir112 // CHECK: "tf.BroadcastGradientArgs"
114 …%r0, %r1 = "tf.BroadcastGradientArgs"(%s0, %s1) {} : (tensor<4xi32>, tensor<4xi32>) -> (tensor<1xi…
Dtf-ops.mlir158 // Test valid tf.BroadcastGradientArgs
161 …%r0, %r1 = "tf.BroadcastGradientArgs"(%s0, %s1) : (tensor<4xi32>, tensor<4xi32>) -> (tensor<1xi32>…
169 …%r0, %r1 = "tf.BroadcastGradientArgs"(%s0, %s1) : (tensor<4xi32>, tensor<4xi64>) -> (tensor<1xi32>…
179 …%r0, %r1 = "tf.BroadcastGradientArgs"(%s0, %s1) : (tensor<2xi32>, tensor<2xi32>) -> (tensor<1xi32>…
189 …%r0, %r1 = "tf.BroadcastGradientArgs"(%s0, %s1) : (tensor<2x2xi32>, tensor<2xi32>) -> (tensor<1xi3…
199 …%r0, %r1 = "tf.BroadcastGradientArgs"(%s0, %s1) : (tensor<2xi32>, tensor<i32>) -> (tensor<1xi32>, …
209 …%r0, %r1 = "tf.BroadcastGradientArgs"(%s0, %s1) : (tensor<2xi32>, tensor<2xi32>) -> (tensor<2x2xi3…
217 …%r0, %r1 = "tf.BroadcastGradientArgs"(%s0, %s1) : (tensor<4xi32>, tensor<4xi32>) -> (tensor<1xi32>…
227 …%r0, %r1 = "tf.BroadcastGradientArgs"(%s0, %s1) : (tensor<2xi32>, tensor<2xi32>) -> (tensor<0xi32>…
237 …%r0, %r1 = "tf.BroadcastGradientArgs"(%s0, %s1) : (tensor<2xi32>, tensor<2xi32>) -> (tensor<0xi32>…
Dshape_inference.mlir87 // CHECK: tf.BroadcastGradientArgs
89 …%2:2 = "tf.BroadcastGradientArgs"(%arg0, %arg1) {T = "tfdtype$DT_INT32", name = "BroadcastGradient…
/external/tensorflow/tensorflow/compiler/jit/tests/
Dopens2s_gnmt_mixed_precision.pbtxt.gz
Dopens2s_gnmt_mixed_precision.golden_summary14 BroadcastGradientArgs 62
86 BroadcastGradientArgs 1
/external/tensorflow/tensorflow/core/api_def/java_api/
Dapi_def_BroadcastGradientArgs.pbtxt2 graph_op_name: "BroadcastGradientArgs"
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_BroadcastGradientArgs.pbtxt2 graph_op_name: "BroadcastGradientArgs"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DBroadcastGradientArgs.pbtxt2 name: "BroadcastGradientArgs"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DBroadcastGradientArgs.pbtxt2 name: "BroadcastGradientArgs"
/external/tensorflow/tensorflow/compiler/mlir/tfr/examples/mnist/
Dops_defs.py90 _, reduction_axes = tf.raw_ops.BroadcastGradientArgs(
154 _, reduction_axes = tf.raw_ops.BroadcastGradientArgs(
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/compile_mlir_util/
Dconstant-folding-hook.mlir8 …%r0, %r1 = "tf.BroadcastGradientArgs"(%0, %0) {T = i32} : (tensor<0xi32>, tensor<0xi32>) -> (tenso…
/external/tensorflow/tensorflow/core/common_runtime/
Dconstant_folding_test.cc247 auto b = ops::internal::BroadcastGradientArgs(s, s0, s1); in TEST_F()
276 auto b = ops::internal::BroadcastGradientArgs(s, s0, s1); in TEST_F()
Dfunction_test.cc1682 auto func1_rx = ops::internal::BroadcastGradientArgs( in TEST_F()
1764 auto rx = ops::internal::BroadcastGradientArgs(s.WithOpName("rx"), sx, sy); in TEST_F()
1794 auto rx = ops::internal::BroadcastGradientArgs(s.WithOpName("rx"), sx, sy); in TEST_F()
1951 auto rx = ops::internal::BroadcastGradientArgs( in TEST_F()
/external/tensorflow/tensorflow/lite/g3doc/guide/
Dop_select_allowlist.md85 * `raw_ops.BroadcastGradientArgs`
/external/tensorflow/tensorflow/compiler/tf2xla/g3doc/
Dgpu_supported_ops.md40 `BroadcastGradientArgs` | `T={int32,int64}`
Dcpu_supported_ops.md40 `BroadcastGradientArgs` | `T={int32,int64}`
/external/tensorflow/tensorflow/cc/gradients/
Dmath_grad.cc368 auto rx = internal::BroadcastGradientArgs(scope, sx_1, sx_2); in BinaryGradCommon()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dconstant_folding_test.cc3042 auto f = ops::internal::BroadcastGradientArgs(s.WithOpName("f"), d, e); in TEST_F()
3049 auto i = ops::internal::BroadcastGradientArgs(s.WithOpName("i"), d, h); in TEST_F()
3125 auto f = ops::internal::BroadcastGradientArgs(s.WithOpName("f"), d, e); in TEST_F()
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc1374 TEST_F(OpTest, BroadcastGradientArgs) { in TEST_F() argument
/external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/
Dlegalization.md1419 ### tf.BroadcastGradientArgs
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.raw_ops.pbtxt648 name: "BroadcastGradientArgs"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.raw_ops.pbtxt648 name: "BroadcastGradientArgs"
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v0.pbtxt7167 name: "BroadcastGradientArgs"
7186 name: "BroadcastGradientArgs"
/external/tensorflow/tensorflow/go/op/
Dwrappers.go1596 func BroadcastGradientArgs(scope *Scope, s0 tf.Output, s1 tf.Output) (r0 tf.Output, r1 tf.Output) { func

12