| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | gather_op.cc | 208 OP_REQUIRES(context, TensorShapeUtils::IsVectorOrHigher(params_shape), in Compile()
210 OP_REQUIRES(context, TensorShapeUtils::IsVectorOrHigher(indices_shape), in Compile()
|
| D | select_op.cc | 55 OP_REQUIRES(ctx, TensorShapeUtils::IsVectorOrHigher(then_shape), in Compile()
|
| D | scatter_nd_op.cc | 91 context, TensorShapeUtils::IsVectorOrHigher(buffer_shape), in Compile()
|
| D | fft_ops.cc | 49 ctx, TensorShapeUtils::IsVectorOrHigher(input_shape), in Compile()
|
| D | softmax_op.cc | 41 OP_REQUIRES(ctx, TensorShapeUtils::IsVectorOrHigher(logits_shape), in Compile()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | gather_nd_op.cc | 84 if (!TensorShapeUtils::IsVectorOrHigher(params.shape())) { in DoGatherNd()
87 if (!TensorShapeUtils::IsVectorOrHigher(indices.shape())) { in DoGatherNd()
|
| D | softmax_op.cc | 64 OP_REQUIRES(context, TensorShapeUtils::IsVectorOrHigher(logits_in.shape()), in Compute()
|
| D | list_kernels.h | 372 c, TensorShapeUtils::IsVectorOrHigher(t.shape()), in Compute()
514 OP_REQUIRES(c, TensorShapeUtils::IsVectorOrHigher(input_tensor.shape()), in Compute()
685 OP_REQUIRES(c, TensorShapeUtils::IsVectorOrHigher(t.shape()), in Compute()
757 OP_REQUIRES(c, TensorShapeUtils::IsVectorOrHigher(input_tensor.shape()), in Compute()
813 OP_REQUIRES(c, TensorShapeUtils::IsVectorOrHigher(input_tensor.shape()), in Compute()
923 OP_REQUIRES(c, TensorShapeUtils::IsVectorOrHigher(input.shape()), in Compute()
|
| D | gather_op.cc | 48 c, TensorShapeUtils::IsVectorOrHigher(params.shape()), in Compute()
|
| D | compare_and_bitpack_op.cc | 50 OP_REQUIRES(c, TensorShapeUtils::IsVectorOrHigher(input_shape), in Compute()
|
| D | matrix_diag_op.cc | 95 OP_REQUIRES(context, TensorShapeUtils::IsVectorOrHigher(input_shape), in Compute()
|
| D | softmax_op_gpu.cu.cc | 137 OP_REQUIRES(context, TensorShapeUtils::IsVectorOrHigher(logits_in_.shape()), in Compute()
|
| D | roll_op.cc | 237 OP_REQUIRES(context, TensorShapeUtils::IsVectorOrHigher(input.shape()), in Compute()
|
| D | cwise_op_select.cc | 90 OP_REQUIRES(ctx, TensorShapeUtils::IsVectorOrHigher(then->shape()), in ComputeBroadcasting()
|
| D | scatter_op.cc | 60 OP_REQUIRES(c, TensorShapeUtils::IsVectorOrHigher(params.shape()), in DoValidationChecking()
|
| D | training_ops.cc | 810 OP_REQUIRES(ctx, TensorShapeUtils::IsVectorOrHigher(var.shape()), in DoCompute()
996 OP_REQUIRES(ctx, TensorShapeUtils::IsVectorOrHigher(var.shape()), in Compute()
1355 OP_REQUIRES(ctx, TensorShapeUtils::IsVectorOrHigher(var.shape()), in Compute()
1496 OP_REQUIRES(ctx, TensorShapeUtils::IsVectorOrHigher(var.shape()), in Compute()
1786 OP_REQUIRES(ctx, TensorShapeUtils::IsVectorOrHigher(var.shape()), in Compute()
2135 OP_REQUIRES(ctx, TensorShapeUtils::IsVectorOrHigher(var.shape()), in Compute()
2501 OP_REQUIRES(ctx, TensorShapeUtils::IsVectorOrHigher(var.shape()), in Compute()
2720 OP_REQUIRES(ctx, TensorShapeUtils::IsVectorOrHigher(var.shape()), in Compute()
3643 OP_REQUIRES(ctx, TensorShapeUtils::IsVectorOrHigher(var.shape()), in Compute()
3781 OP_REQUIRES(ctx, TensorShapeUtils::IsVectorOrHigher(var.shape()), in Compute()
|
| D | tensor_array_ops.cc | 888 ctx, TensorShapeUtils::IsVectorOrHigher(value_shape_t), in Compute()
1268 ctx, TensorShapeUtils::IsVectorOrHigher(tensor_value->shape()), in Compute()
|
| D | scatter_nd_op.cc | 561 if (!TensorShapeUtils::IsVectorOrHigher(params_shape)) { in PrepareAndValidateInputs()
|
| D | resource_variable_ops.cc | 630 c, TensorShapeUtils::IsVectorOrHigher(params.shape()), in Compute()
|
| /external/tensorflow/tensorflow/contrib/coder/kernels/ |
| D | pmf_to_cdf_op.cc | 52 OP_REQUIRES(context, TensorShapeUtils::IsVectorOrHigher(shape), in Compute()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor_shape.h | 360 static bool IsVectorOrHigher(const TensorShape& shape) { in IsVectorOrHigher() function
|