| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| hexagon_rewriter_transform.cc | 54 string shape_string; local
56 INPUT_SHAPE_PREFIX + std::to_string(i), "", &shape_string));
58 CHECK(str_util::SplitAndParseAsInts(shape_string, ',', &dims));
69 << ", shape = " << shape_string
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| text_literal_reader.cc | 87 string shape_string; local
88 Status s = buf.ReadLine(&shape_string);
93 tensorflow::StringPiece sp(shape_string);
96 shape_string = tmp;
98 TF_ASSIGN_OR_RETURN(Shape shape, ShapeUtil::ParseShapeString(shape_string));
|
| shape_util_test.cc | 74 string shape_string = "f32[123,456]"; local
76 ShapeUtil::ParseShapeString(shape_string));
84 string shape_string = "(f32[1572864],s8[5120,1024])"; local
86 ShapeUtil::ParseShapeString(shape_string));
96 string shape_string = "(f32[1],(f32[2]), f32[3])"; local
98 ShapeUtil::ParseShapeString(shape_string));
110 string shape_string = "f32[123,456]{0,1}"; local
112 ShapeUtil::ParseShapeString(shape_string));
120 string shape_string = "f32[123,456]dense{0,1}"; local
122 ShapeUtil::ParseShapeString(shape_string));
130 string shape_string = "f32[123,456]sparse{10}"; local
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_cse.cc | 66 string shape_string = shape.ShortDebugString(); local
69 auto range = constants.equal_range(shape_string);
78 constants.emplace(shape_string, instruction);
|
| shape_inference.cc | 2192 auto shape_string = [&]() { local
2261 auto shape_string = [&]() { local
[all...] |
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| strip_unused_nodes.cc | 89 const string& shape_string = context.params.at("shape")[0]; local
90 TF_RETURN_IF_ERROR(TensorShapeFromString(shape_string, result));
107 const string& shape_string = context.params.at("shape_for_name")[i]; local
108 TF_RETURN_IF_ERROR(TensorShapeFromString(shape_string, result));
|
| transform_utils.cc | 589 Status TensorShapeFromString(const string& shape_string, TensorShape* result) {
590 if (shape_string.empty()) {
594 if (!str_util::SplitAndParseAsInts(shape_string, ',', &dims)) {
595 return errors::InvalidArgument("Could parse as shape: '", shape_string,
|
| transform_utils.h | 137 Status TensorShapeFromString(const string& shape_string, TensorShape* result);
|
