| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| select_op.cc | 55 OP_REQUIRES(ctx, TensorShapeUtils::IsVectorOrHigher(then_shape),
|
| scatter_nd_op.cc | 91 context, TensorShapeUtils::IsVectorOrHigher(buffer_shape),
|
| gather_op.cc | 208 OP_REQUIRES(context, TensorShapeUtils::IsVectorOrHigher(params_shape),
210 OP_REQUIRES(context, TensorShapeUtils::IsVectorOrHigher(indices_shape),
|
| fft_ops.cc | 49 ctx, TensorShapeUtils::IsVectorOrHigher(input_shape),
|
| softmax_op.cc | 41 OP_REQUIRES(ctx, TensorShapeUtils::IsVectorOrHigher(logits_shape),
|
| /external/tensorflow/tensorflow/core/kernels/ |
| softmax_op.cc | 64 OP_REQUIRES(context, TensorShapeUtils::IsVectorOrHigher(logits_in.shape()),
|
| gather_nd_op.cc | 84 if (!TensorShapeUtils::IsVectorOrHigher(params.shape())) {
87 if (!TensorShapeUtils::IsVectorOrHigher(indices.shape())) {
|
| gather_op.cc | 48 c, TensorShapeUtils::IsVectorOrHigher(params.shape()),
|
| list_kernels.h | 372 c, TensorShapeUtils::IsVectorOrHigher(t.shape()),
514 OP_REQUIRES(c, TensorShapeUtils::IsVectorOrHigher(input_tensor.shape()),
685 OP_REQUIRES(c, TensorShapeUtils::IsVectorOrHigher(t.shape()),
757 OP_REQUIRES(c, TensorShapeUtils::IsVectorOrHigher(input_tensor.shape()),
[all...] |
| compare_and_bitpack_op.cc | 50 OP_REQUIRES(c, TensorShapeUtils::IsVectorOrHigher(input_shape),
|
| matrix_diag_op.cc | 95 OP_REQUIRES(context, TensorShapeUtils::IsVectorOrHigher(input_shape),
|
| softmax_op_gpu.cu.cc | 137 OP_REQUIRES(context, TensorShapeUtils::IsVectorOrHigher(logits_in_.shape()),
|
| cwise_op_select.cc | 90 OP_REQUIRES(ctx, TensorShapeUtils::IsVectorOrHigher(then->shape()),
|
| roll_op.cc | 237 OP_REQUIRES(context, TensorShapeUtils::IsVectorOrHigher(input.shape()),
|
| scatter_op.cc | 60 OP_REQUIRES(c, TensorShapeUtils::IsVectorOrHigher(params.shape()),
|
| training_ops.cc | 810 OP_REQUIRES(ctx, TensorShapeUtils::IsVectorOrHigher(var.shape())
[all...] |
| tensor_array_ops.cc | [all...] |
| resource_variable_ops.cc | 630 c, TensorShapeUtils::IsVectorOrHigher(params.shape()),
[all...] |
| scatter_nd_op.cc | 561 if (!TensorShapeUtils::IsVectorOrHigher(params_shape)) {
[all...] |
| /external/tensorflow/tensorflow/contrib/coder/kernels/ |
| pmf_to_cdf_op.cc | 52 OP_REQUIRES(context, TensorShapeUtils::IsVectorOrHigher(shape),
|
| /external/tensorflow/tensorflow/core/framework/ |
| tensor_shape.h | 360 static bool IsVectorOrHigher(const TensorShape& shape) {
|
