| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| matrix_band_part_op.cc | 46 errors::InvalidArgument("num_upper must be scalar, got shape ",
52 xla::XlaOp num_upper = context->Input(2); variable
70 // If num_lower or num_upper are negative, include all lower/upper
76 num_upper = xla::Select(xla::Lt(num_upper, zero_index),
78 num_upper);
81 xla::Le(offset, num_upper));
|
| /external/tensorflow/tensorflow/core/kernels/ |
| matrix_band_part_op.cc | 82 errors::InvalidArgument("num_upper must be scalar, got shape ",
84 const int64 num_upper = as_int64_scalar(num_upper_in); variable
85 OP_REQUIRES(context, num_upper <= input_reshaped.dimension(2),
86 errors::InvalidArgument("num_upper must be negative or less or "
89 ") got: ", num_upper));
93 (num_upper < 0 || num_upper == input_reshaped.dimension(2)))) {
104 fn(context, context->eigen_device<Device>(), num_lower, num_upper,
225 .HostMemory("num_upper"), \
240 .HostMemory("num_upper"), \
[all...] |
| /external/tensorflow/tensorflow/cc/gradients/ |
| array_grad.cc | 169 auto num_upper = op.input(2); local
171 MatrixBandPart(scope, grad_inputs[0], num_lower, num_upper));
|
| array_grad_test.cc | 147 const int64 num_upper = 2; local
148 auto y = MatrixBandPart(scope_, x, num_lower, num_upper);
|
| /external/tensorflow/tensorflow/python/ops/ |
| array_grad.py | 368 num_upper = op.inputs[2]
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...] |
