Searched refs:shape_error (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | clip_by_value_op.cc | 37 auto shape_error = [&]() -> tensorflow::Status { in Compile() local
47 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(min_shape), shape_error()); in Compile()
51 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(max_shape), shape_error()); in Compile()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | canonicalize.td | 151 (TF_LogicalNotOp (TF_EqualOp $arg0, $arg1, $shape_error)),
152 (TF_NotEqualOp $arg0, $arg1, $shape_error)>;
155 (TF_LogicalNotOp (TF_NotEqualOp $arg0, $arg1, $shape_error)),
156 (TF_EqualOp $arg0, $arg1, $shape_error)>;
|
| /external/tensorflow/tensorflow/lite/kernels/parse_example/ |
| D | parse_example.cc | 185 auto shape_error = [&](size_t size, StringPiece type_str) { in FastParseSerializedExample() local
200 return shape_error(num_elements - slice.EndDistance(), "int64"); in FastParseSerializedExample()
209 return shape_error(num_elements - slice.EndDistance(), "float"); in FastParseSerializedExample()
219 return shape_error(num_elements - slice.EndDistance(), "bytes"); in FastParseSerializedExample()
239 auto shape_error = [&](size_t size, StringPiece type_str) { in FastParseSerializedExample() local
254 return shape_error(out.int64_list.size(), "int64"); in FastParseSerializedExample()
266 return shape_error(out.float_list.size(), "float"); in FastParseSerializedExample()
278 return shape_error(out.bytes_list.size(), "byte"); in FastParseSerializedExample()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | example_proto_fast_parsing.cc | 672 auto shape_error = [&](size_t size, StringPiece type_str) { in FastParseSerializedExample() local
687 return shape_error(num_elements - slice.EndDistance(), "int64"); in FastParseSerializedExample()
696 return shape_error(num_elements - slice.EndDistance(), "float"); in FastParseSerializedExample()
705 return shape_error(num_elements - slice.EndDistance(), "bytes"); in FastParseSerializedExample()
724 auto shape_error = [&](size_t size, StringPiece type_str) { in FastParseSerializedExample() local
739 return shape_error(out.int64_list.size(), "int64"); in FastParseSerializedExample()
751 return shape_error(out.float_list.size(), "float"); in FastParseSerializedExample()
763 return shape_error(out.bytes_list.size(), "bytes"); in FastParseSerializedExample()
|