Home
last modified time | relevance | path

Searched refs:shape_error (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dclip_by_value_op.cc37 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/core/util/
Dexample_proto_fast_parsing.cc625 auto shape_error = [&](size_t size, StringPiece type_str) { in FastParseSerializedExample() local
640 return shape_error(num_elements - slice.EndDistance(), "int64"); in FastParseSerializedExample()
649 return shape_error(num_elements - slice.EndDistance(), "float"); in FastParseSerializedExample()
658 return shape_error(num_elements - slice.EndDistance(), "bytes"); in FastParseSerializedExample()
677 auto shape_error = [&](size_t size, StringPiece type_str) { in FastParseSerializedExample() local
692 return shape_error(out.int64_list.size(), "int64"); in FastParseSerializedExample()
704 return shape_error(out.float_list.size(), "float"); in FastParseSerializedExample()
716 return shape_error(out.bytes_list.size(), "bytes"); in FastParseSerializedExample()