| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| matrix_band_part_op.cc | 41 errors::InvalidArgument("num_lower must be scalar, got shape ",
51 xla::XlaOp num_lower = context->Input(1); variable
70 // If num_lower or num_upper are negative, include all lower/upper
73 num_lower = xla::Select(xla::Lt(num_lower, zero_index),
75 num_lower);
80 auto indicator = xla::And(xla::Le(xla::Neg(num_lower), offset),
|
| /external/tensorflow/tensorflow/core/kernels/ |
| matrix_band_part_op.cc | 63 errors::InvalidArgument("num_lower must be scalar, got shape ",
73 const int64 num_lower = as_int64_scalar(num_lower_in); variable
75 context, num_lower <= input_reshaped.dimension(1),
77 "num_lower must be negative or less or equal to number of rows (",
78 input_reshaped.dimension(1), ") got: ", num_lower));
92 ((num_lower < 0 || num_lower == input_reshaped.dimension(1)) &&
104 fn(context, context->eigen_device<Device>(), num_lower, num_upper,
224 .HostMemory("num_lower") \
239 .HostMemory("num_lower") \
[all...] |
| /external/tensorflow/tensorflow/cc/gradients/ |
| array_grad.cc | 168 auto num_lower = op.input(1); local
171 MatrixBandPart(scope, grad_inputs[0], num_lower, num_upper));
|
| array_grad_test.cc | 146 const int64 num_lower = 1; local
148 auto y = MatrixBandPart(scope_, x, num_lower, num_upper);
|
| /external/tensorflow/tensorflow/python/ops/ |
| array_grad.py | 367 num_lower = op.inputs[1]
369 return (array_ops.matrix_band_part(grad, num_lower, num_upper), None, None)
[all...] |
| /external/tensorflow/tensorflow/python/ops/distributions/ |
| util.py | [all...] |
| /external/tensorflow/tensorflow/go/op/ |
| wrappers.go | [all...] |
