Home
last modified time | relevance | path

Searched refs:IsMatrixOrHigher (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dbias_ops.cc45 OP_REQUIRES(ctx, TensorShapeUtils::IsMatrixOrHigher(input_shape), in Compile()
93 OP_REQUIRES(ctx, TensorShapeUtils::IsMatrixOrHigher(out_backprop_shape), in Compile()
Dmatrix_band_part_op.cc34 OP_REQUIRES(context, TensorShapeUtils::IsMatrixOrHigher(input_shape), in Compile()
Dmatrix_diag_ops.cc360 OP_REQUIRES(context, TensorShapeUtils::IsMatrixOrHigher(input_shape), in Compile()
472 OP_REQUIRES(context, TensorShapeUtils::IsMatrixOrHigher(input_shape), in Compile()
/external/tensorflow/tensorflow/core/kernels/
Dbias_op.cc103 OP_REQUIRES(context, TensorShapeUtils::IsMatrixOrHigher(input.shape()), in Compute()
234 TensorShapeUtils::IsMatrixOrHigher(output_backprop.shape()), in Compute()
314 OP_REQUIRES(context, TensorShapeUtils::IsMatrixOrHigher(input.shape()), in Compute()
505 TensorShapeUtils::IsMatrixOrHigher(output_backprop.shape()), in Compute()
Dquantized_bias_add_op.cc46 OP_REQUIRES(context, TensorShapeUtils::IsMatrixOrHigher(input.shape()), in Compute()
Dencode_proto_op.cc553 ctx, TensorShapeUtils::IsMatrixOrHigher(v.shape()), in Compute()
Dexample_parsing_ops.cc132 if (TensorShapeUtils::IsMatrixOrHigher(serialized->shape())) { in CheckInputShapes()
496 if (TensorShapeUtils::IsMatrixOrHigher(serialized->shape())) { in CheckInputShapes()
/external/tensorflow/tensorflow/core/kernels/linalg/
Dmatrix_band_part_op.cc56 OP_REQUIRES(context, TensorShapeUtils::IsMatrixOrHigher(input_shape), in Compute()
Dmatrix_set_diag_op.cc93 OP_REQUIRES(context, TensorShapeUtils::IsMatrixOrHigher(input_shape), in Compute()
Dmatrix_diag_op.cc94 OP_REQUIRES(context, TensorShapeUtils::IsMatrixOrHigher(input_shape), in Compute()
/external/tensorflow/tensorflow/core/framework/
Dtensor_shape.h473 static bool IsMatrixOrHigher(const TensorShape& shape) { in IsMatrixOrHigher() function