| /external/tensorflow/tensorflow/core/kernels/ |
| D | gather_nd_op.h | 51 if (!TensorShapeUtils::IsVectorOrHigher(params.shape())) { in DoGatherNd()
54 if (!TensorShapeUtils::IsVectorOrHigher(indices.shape())) { in DoGatherNd()
|
| D | softmax_op.cc | 57 OP_REQUIRES(context, TensorShapeUtils::IsVectorOrHigher(logits_in.shape()), in Compute()
|
| D | list_kernels.h | 348 c, TensorShapeUtils::IsVectorOrHigher(t.shape()), in Compute()
491 OP_REQUIRES(c, TensorShapeUtils::IsVectorOrHigher(input_tensor.shape()), in Compute()
662 OP_REQUIRES(c, TensorShapeUtils::IsVectorOrHigher(t.shape()), in Compute()
734 OP_REQUIRES(c, TensorShapeUtils::IsVectorOrHigher(input_tensor.shape()), in Compute()
790 OP_REQUIRES(c, TensorShapeUtils::IsVectorOrHigher(input_tensor.shape()), in Compute()
900 OP_REQUIRES(c, TensorShapeUtils::IsVectorOrHigher(input.shape()), in Compute()
|
| D | fingerprint_op.cc | 89 context, TensorShapeUtils::IsVectorOrHigher(input.shape()), in Compute()
|
| D | segment_reduction_ops_impl_1.cc | 25 OP_REQUIRES(context, TensorShapeUtils::IsVectorOrHigher(input.shape()), in SegmentReductionValidationHelper()
|
| D | compare_and_bitpack_op.cc | 50 OP_REQUIRES(c, TensorShapeUtils::IsVectorOrHigher(input_shape), in Compute()
|
| D | scatter_op.cc | 54 OP_REQUIRES(c, TensorShapeUtils::IsVectorOrHigher(params.shape()), in DoValidationChecking()
|
| D | gather_op.cc | 58 c, TensorShapeUtils::IsVectorOrHigher(params.shape()), in Compute()
|
| D | softmax_op_gpu.cu.cc | 203 OP_REQUIRES(context, TensorShapeUtils::IsVectorOrHigher(logits_in_.shape()), in Compute()
|
| D | roll_op.cc | 48 OP_REQUIRES(context, TensorShapeUtils::IsVectorOrHigher(input.shape()), in Compute()
|
| D | cwise_op_select.cc | 84 OP_REQUIRES(ctx, TensorShapeUtils::IsVectorOrHigher(then->shape()), in ComputeBroadcasting()
|
| D | training_ops.cc | 1217 OP_REQUIRES(ctx, TensorShapeUtils::IsVectorOrHigher(var.shape()), in DoCompute()
1401 OP_REQUIRES(ctx, TensorShapeUtils::IsVectorOrHigher(var.shape()), in Compute()
1880 OP_REQUIRES(ctx, TensorShapeUtils::IsVectorOrHigher(var.shape()), in Compute()
2010 OP_REQUIRES(ctx, TensorShapeUtils::IsVectorOrHigher(var.shape()), in Compute()
2143 OP_REQUIRES(ctx, TensorShapeUtils::IsVectorOrHigher(var.shape()), in Compute()
2402 OP_REQUIRES(ctx, TensorShapeUtils::IsVectorOrHigher(var.shape()), in Compute()
2810 OP_REQUIRES(ctx, TensorShapeUtils::IsVectorOrHigher(var.shape()), in Compute()
3173 OP_REQUIRES(ctx, TensorShapeUtils::IsVectorOrHigher(var.shape()), in Compute()
3400 OP_REQUIRES(ctx, TensorShapeUtils::IsVectorOrHigher(var.shape()), in Compute()
4242 OP_REQUIRES(ctx, TensorShapeUtils::IsVectorOrHigher(var.shape()), in Compute()
[all …]
|
| D | tensor_array_ops.cc | 882 ctx, TensorShapeUtils::IsVectorOrHigher(value_shape_t), in Compute()
1260 ctx, TensorShapeUtils::IsVectorOrHigher(tensor_value->shape()), in Compute()
|
| D | resource_variable_ops.cc | 634 c, TensorShapeUtils::IsVectorOrHigher(params.shape()), in Compute()
|
| D | scatter_nd_op.cc | 825 if (!TensorShapeUtils::IsVectorOrHigher(params_shape)) { in PrepareAndValidateInputs()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | scatter_nd_op.cc | 93 context, TensorShapeUtils::IsVectorOrHigher(buffer_shape), in Compile()
147 context, TensorShapeUtils::IsVectorOrHigher(buffer_shape), in CompileTensorScatter()
|
| D | gather_op.cc | 277 OP_REQUIRES(context, TensorShapeUtils::IsVectorOrHigher(params_shape), in Compile()
279 OP_REQUIRES(context, TensorShapeUtils::IsVectorOrHigher(indices_shape), in Compile()
|
| D | select_op.cc | 57 OP_REQUIRES(ctx, TensorShapeUtils::IsVectorOrHigher(then_shape), in Compile()
|
| D | fft_ops.cc | 49 ctx, TensorShapeUtils::IsVectorOrHigher(input_shape), in Compile()
|
| D | matrix_diag_ops.cc | 245 OP_REQUIRES(context, TensorShapeUtils::IsVectorOrHigher(diag_shape), in Compile()
476 OP_REQUIRES(context, TensorShapeUtils::IsVectorOrHigher(diag_shape), in Compile()
|
| D | softmax_op.cc | 57 OP_REQUIRES(ctx, TensorShapeUtils::IsVectorOrHigher(logits_shape), in Compile()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| D | get_item_op.cc | 39 ctx, TensorShapeUtils::IsVectorOrHigher(data_shape), in Compile()
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | matrix_set_diag_op.cc | 97 OP_REQUIRES(context, TensorShapeUtils::IsVectorOrHigher(diag_shape), in Compute()
|
| D | matrix_diag_op.cc | 204 OP_REQUIRES(context, TensorShapeUtils::IsVectorOrHigher(diagonal_shape), in Compute()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor_shape.h | 463 static bool IsVectorOrHigher(const TensorShape& shape) { in IsVectorOrHigher() function
|