Searched refs:IsMatrixOrHigher (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | bias_ops.cc | 45 OP_REQUIRES(ctx, TensorShapeUtils::IsMatrixOrHigher(input_shape), in Compile()
93 OP_REQUIRES(ctx, TensorShapeUtils::IsMatrixOrHigher(out_backprop_shape), in Compile()
|
| D | matrix_set_diag_op.cc | 35 OP_REQUIRES(context, TensorShapeUtils::IsMatrixOrHigher(input_shape), in Compile()
|
| D | matrix_band_part_op.cc | 34 OP_REQUIRES(context, TensorShapeUtils::IsMatrixOrHigher(input_shape), in Compile()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | bias_op.cc | 104 OP_REQUIRES(context, TensorShapeUtils::IsMatrixOrHigher(input.shape()), in Compute()
249 TensorShapeUtils::IsMatrixOrHigher(output_backprop.shape()), in Compute()
347 OP_REQUIRES(context, TensorShapeUtils::IsMatrixOrHigher(input.shape()), in Compute()
537 TensorShapeUtils::IsMatrixOrHigher(output_backprop.shape()), in Compute()
|
| D | quantized_bias_add_op.cc | 46 OP_REQUIRES(context, TensorShapeUtils::IsMatrixOrHigher(input.shape()), in Compute()
|
| D | matrix_set_diag_op.cc | 56 OP_REQUIRES(context, TensorShapeUtils::IsMatrixOrHigher(input_shape), in Compute()
|
| D | matrix_diag_op.cc | 56 OP_REQUIRES(context, TensorShapeUtils::IsMatrixOrHigher(input_shape), in Compute()
|
| D | matrix_band_part_op.cc | 55 OP_REQUIRES(context, TensorShapeUtils::IsMatrixOrHigher(input_shape), in Compute()
|
| D | encode_proto_op.cc | 533 ctx, TensorShapeUtils::IsMatrixOrHigher(v.shape()), in Compute()
|
| /external/tensorflow/tensorflow/contrib/resampler/kernels/ |
| D | resampler_ops.cc | 144 ::tensorflow::TensorShapeUtils::IsMatrixOrHigher(warp_shape), in Compute()
354 ::tensorflow::TensorShapeUtils::IsMatrixOrHigher(warp_shape), in Compute()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor_shape.h | 370 static bool IsMatrixOrHigher(const TensorShape& shape) { in IsMatrixOrHigher() function
|