| /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/ |
| D | matrix_band_part_op.cc | 84 const int64 num_upper = as_int64_scalar(num_upper_in); in Compute() local
85 OP_REQUIRES(context, num_upper <= input_reshaped.dimension(2), in Compute()
89 ") got: ", num_upper)); in Compute()
93 (num_upper < 0 || num_upper == input_reshaped.dimension(2)))) { 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 | 17 name: "num_upper"
42 (num_upper < 0 || (n-m) <= 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 | 169 auto num_upper = op.input(2); 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 | 368 num_upper = op.inputs[2]
369 return (array_ops.matrix_band_part(grad, num_lower, num_upper), None, None)
|
| /external/tensorflow/tensorflow/python/ops/distributions/ |
| D | util.py | 902 num_upper=(-1 if upper else 0))
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v0.pbtxt | 4787 name: "num_upper"
4810 name: "num_upper"
16014 name: "num_upper"
|
| D | ops_history.v2.pbtxt | 9408 name: "num_upper"
34801 name: "num_upper"
34824 name: "num_upper"
|
| D | ops_history.v1.pbtxt | 9408 name: "num_upper"
35239 name: "num_upper"
35262 name: "num_upper"
|
| /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 | 3362 name: "num_upper"
18286 name: "num_upper"
|