| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | matrix_band_part_op.cc | 51 xla::XlaOp num_lower = context->Input(1); in Compile() local
73 num_lower = xla::Select(xla::Lt(num_lower, zero_index), in Compile()
75 num_lower); in Compile()
80 auto indicator = xla::And(xla::Le(xla::Neg(num_lower), offset), in Compile()
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | matrix_band_part_op.cc | 74 const int64 num_lower = as_int64_scalar(num_lower_in); in Compute() local
76 context, num_lower <= input_reshaped.dimension(1), in Compute()
79 input_reshaped.dimension(1), ") got: ", num_lower)); in Compute()
93 ((num_lower < 0 || num_lower == input_reshaped.dimension(1)) && 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 | 10 name: "num_lower"
39 `in_band(m, n) = (num_lower < 0 || (m-n) <= num_lower)) &&
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | MatrixBandPart.pbtxt | 8 name: "num_lower"
31 name: "num_lower"
|
| D | BatchMatrixBandPart.pbtxt | 8 name: "num_lower"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | MatrixBandPart.pbtxt | 8 name: "num_lower"
31 name: "num_lower"
|
| D | BatchMatrixBandPart.pbtxt | 8 name: "num_lower"
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_patterns.td | 401 // and two integers, `num_lower` and `num_upper`:
405 // num_lower_or_m = (num_lower < 0) ? m : num_lower
413 def : Pattern<(TF_MatrixBandPartOp:$op AnyStaticShapeTensor:$input, $num_lower,
415 [(HLO_ConstOp:$m_dim (GetDimensionSizeFromEnd<"1"> $input, $num_lower)),
419 $num_lower, (HLO_ConstOp:$zero (ConstantSplat<"0"> $num_lower)),
423 $num_lower
438 (createIotaOp<"1"> $op, $input, $num_lower),
439 (createIotaOp<"0"> $op, $input, $num_lower)
|
| D | legalize_tf.cc | 790 static RankedTensorType Get2DTensorType(Value input, Value num_lower) { in Get2DTensorType() argument
794 auto element_type = num_lower.getType().cast<TensorType>().getElementType(); in Get2DTensorType()
|
| /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 | 2136 num_lower = pfor_input.unstacked_input(1)
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 | 146 const int64 num_lower = 1; in TEST_F() local
148 auto y = MatrixBandPart(scope_, x, num_lower, num_upper); in TEST_F()
|
| D | array_grad.cc | 177 auto num_lower = op.input(1); 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 | 529 num_lower = op.inputs[1]
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 | 2675 if (!IsOfRankOrUnranked(op.num_lower(), 0)) { in Verify()
2678 << op.num_lower().getType(); in Verify()
|
| D | tf_generated_ops.td | 7144 `in_band(m, n) = (num_lower < 0 || (m-n) <= num_lower)) &&
7178 lower triangle.}]>:$num_lower,
|
| /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 | 4783 name: "num_lower"
4806 name: "num_lower"
16010 name: "num_lower"
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | tf-ops.mlir | 1415 // expected-error @+1 {{op requires `num_lower` 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,
|