| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | matrix_band_part_op.cc | 52 xla::XlaOp num_upper = context->Input(2); in Compile() local
76 num_upper = xla::Select(xla::Lt(num_upper, zero_index), in Compile()
78 num_upper); in Compile()
81 xla::Le(offset, num_upper)); in Compile()
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | matrix_band_part_op.cc | 85 const int64 num_upper = as_int64_scalar(num_upper_in); in Compute() local
86 OP_REQUIRES(context, num_upper <= input_reshaped.dimension(2), in Compute()
90 ") got: ", num_upper)); in Compute()
94 (num_upper < 0 || num_upper == input_reshaped.dimension(2)))) { in Compute()
105 fn(context, context->eigen_device<Device>(), num_lower, num_upper, in Compute()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_MatrixBandPart.pbtxt | 17 name: "num_upper"
40 (num_upper < 0 || (n-m) <= num_upper)`.
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | MatrixBandPart.pbtxt | 12 name: "num_upper"
35 name: "num_upper"
|
| D | BatchMatrixBandPart.pbtxt | 12 name: "num_upper"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | MatrixBandPart.pbtxt | 12 name: "num_upper"
35 name: "num_upper"
|
| D | BatchMatrixBandPart.pbtxt | 12 name: "num_upper"
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_patterns.td | 401 // and two integers, `num_lower` and `num_upper`:
406 // num_upper_or_n = (num_upper < 0) ? n : num_upper
414 $num_upper),
416 (HLO_ConstOp:$n_dim (GetDimensionSizeFromEnd<"0"> $input, $num_upper)),
427 $num_upper, $zero, HLO_COMPARISON_DIRECTION_LT,
431 $num_upper
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | array_test.py | 368 for num_lower, num_upper in ((0, -1), (-1, 0), (1, 1)):
372 array_ops.gather(x, i), num_lower=num_lower, num_upper=num_upper)
|
| D | pfor.py | 2137 num_upper = pfor_input.unstacked_input(2)
2139 array_ops.matrix_band_part(t, num_lower=num_lower, num_upper=num_upper),
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | array_grad_test.cc | 147 const int64 num_upper = 2; in TEST_F() local
148 auto y = MatrixBandPart(scope_, x, num_lower, num_upper); in TEST_F()
|
| D | array_grad.cc | 178 auto num_upper = op.input(2); in MatrixBandPartGrad() local
180 MatrixBandPart(scope, grad_inputs[0], num_lower, num_upper)); in MatrixBandPartGrad()
|
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linalg_impl.py | 1004 band_part(lower_upper, num_lower=-1, num_upper=0),
1127 band_part(lower_upper, num_lower=-1, num_upper=0),
1129 upper = band_part(lower_upper, num_lower=0, num_upper=-1)
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.linalg.pbtxt | 97 …argspec: "args=[\'input\', \'num_lower\', \'num_upper\', \'name\'], varargs=None, keywords=None, d…
|
| D | tensorflow.pbtxt | 1641 …argspec: "args=[\'input\', \'num_lower\', \'num_upper\', \'name\'], varargs=None, keywords=None, d…
|
| D | tensorflow.raw_ops.pbtxt | 373 …argspec: "args=[\'input\', \'num_lower\', \'num_upper\', \'name\'], varargs=None, keywords=None, d…
2373 …argspec: "args=[\'input\', \'num_lower\', \'num_upper\', \'name\'], varargs=None, keywords=None, d…
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_grad.py | 530 num_upper = op.inputs[2]
531 return (array_ops.matrix_band_part(grad, num_lower, num_upper), None, None)
|
| /external/tensorflow/tensorflow/python/ops/distributions/ |
| D | util.py | 908 x, num_lower=(0 if upper else -1), num_upper=(-1 if upper else 0))
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_a_m.cc | 2680 if (!IsOfRankOrUnranked(op.num_upper(), 0)) { in Verify()
2683 << op.num_upper().getType(); in Verify()
|
| D | tf_generated_ops.td | 7145 (num_upper < 0 || (n-m) <= num_upper)`.
7180 entire upper triangle.}]>:$num_upper
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 373 …argspec: "args=[\'input\', \'num_lower\', \'num_upper\', \'name\'], varargs=None, keywords=None, d…
2373 …argspec: "args=[\'input\', \'num_lower\', \'num_upper\', \'name\'], varargs=None, keywords=None, d…
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v0.pbtxt | 4787 name: "num_upper"
4810 name: "num_upper"
16014 name: "num_upper"
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | tf-ops.mlir | 1424 // expected-error @+1 {{op requires `num_upper` to have 0 dimensions, but found 'tensor<32xi64>'}}
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 7064 func MatrixBandPart(scope *Scope, input tf.Output, num_lower tf.Output, num_upper tf.Output) (band …
7071 input, num_lower, num_upper,
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 3770 name: "num_upper"
23519 name: "num_upper"
|