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_band_part_op.cc | 34 OP_REQUIRES(context, TensorShapeUtils::IsMatrixOrHigher(input_shape), in Compile()
|
D | matrix_diag_ops.cc | 360 OP_REQUIRES(context, TensorShapeUtils::IsMatrixOrHigher(input_shape), in Compile() 472 OP_REQUIRES(context, TensorShapeUtils::IsMatrixOrHigher(input_shape), in Compile()
|
/external/tensorflow/tensorflow/core/kernels/ |
D | bias_op.cc | 103 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()
|
D | quantized_bias_add_op.cc | 46 OP_REQUIRES(context, TensorShapeUtils::IsMatrixOrHigher(input.shape()), in Compute()
|
D | encode_proto_op.cc | 553 ctx, TensorShapeUtils::IsMatrixOrHigher(v.shape()), in Compute()
|
D | example_parsing_ops.cc | 132 if (TensorShapeUtils::IsMatrixOrHigher(serialized->shape())) { in CheckInputShapes() 496 if (TensorShapeUtils::IsMatrixOrHigher(serialized->shape())) { in CheckInputShapes()
|
/external/tensorflow/tensorflow/core/kernels/linalg/ |
D | matrix_band_part_op.cc | 56 OP_REQUIRES(context, TensorShapeUtils::IsMatrixOrHigher(input_shape), in Compute()
|
D | matrix_set_diag_op.cc | 93 OP_REQUIRES(context, TensorShapeUtils::IsMatrixOrHigher(input_shape), in Compute()
|
D | matrix_diag_op.cc | 94 OP_REQUIRES(context, TensorShapeUtils::IsMatrixOrHigher(input_shape), in Compute()
|
/external/tensorflow/tensorflow/core/framework/ |
D | tensor_shape.h | 473 static bool IsMatrixOrHigher(const TensorShape& shape) { in IsMatrixOrHigher() function
|