| /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/ |
| D | matrix_band_part_op.cc | 73 const int64 num_lower = as_int64_scalar(num_lower_in); in Compute() local
75 context, num_lower <= input_reshaped.dimension(1), in Compute()
78 input_reshaped.dimension(1), ") got: ", num_lower)); in Compute()
92 ((num_lower < 0 || num_lower == input_reshaped.dimension(1)) && in Compute()
104 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"
41 `in_band(m, n) = (num_lower < 0 || (m-n) <= num_lower)) &&
|
| /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 | 168 auto num_lower = op.input(1); in MatrixBandPartGrad() local
171 MatrixBandPart(scope, grad_inputs[0], num_lower, num_upper)); in MatrixBandPartGrad()
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.linalg.pbtxt | 73 …argspec: "args=[\'input\', \'num_lower\', \'num_upper\', \'name\'], varargs=None, keywords=None, d…
|
| D | tensorflow.raw_ops.pbtxt | 321 …argspec: "args=[\'input\', \'num_lower\', \'num_upper\', \'name\'], varargs=None, keywords=None, d…
1901 …argspec: "args=[\'input\', \'num_lower\', \'num_upper\', \'name\'], varargs=None, keywords=None, d…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.linalg.pbtxt | 73 …argspec: "args=[\'input\', \'num_lower\', \'num_upper\', \'name\'], varargs=None, keywords=None, d…
|
| D | tensorflow.pbtxt | 1577 …argspec: "args=[\'input\', \'num_lower\', \'num_upper\', \'name\'], varargs=None, keywords=None, d…
|
| D | tensorflow.raw_ops.pbtxt | 321 …argspec: "args=[\'input\', \'num_lower\', \'num_upper\', \'name\'], varargs=None, keywords=None, d…
1901 …argspec: "args=[\'input\', \'num_lower\', \'num_upper\', \'name\'], varargs=None, keywords=None, d…
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_grad.py | 367 num_lower = op.inputs[1]
369 return (array_ops.matrix_band_part(grad, num_lower, num_upper), None, None)
|
| /external/tensorflow/tensorflow/python/ops/distributions/ |
| D | util.py | 901 num_lower=(0 if upper else -1),
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v0.pbtxt | 4783 name: "num_lower"
4806 name: "num_lower"
16010 name: "num_lower"
|
| D | ops_history.v2.pbtxt | 9404 name: "num_lower"
34797 name: "num_lower"
34820 name: "num_lower"
|
| D | ops_history.v1.pbtxt | 9404 name: "num_lower"
35235 name: "num_lower"
35258 name: "num_lower"
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 7530 func MatrixBandPart(scope *Scope, input tf.Output, num_lower tf.Output, num_upper tf.Output) (band …
7537 input, num_lower, num_upper,
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 3358 name: "num_lower"
18282 name: "num_lower"
|