Searched refs:input_shapes_vector (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| H A D | tf_xla_mlir_translate.cc | 135 std::vector<llvm::Optional<std::vector<int>>> input_shapes_vector; in ParseArgumentShapes() local
136 TF_RETURN_IF_ERROR(ParseNodeShapes(input_shapes_str, input_shapes_vector)); in ParseArgumentShapes()
137 arg_shapes.resize(input_shapes_vector.size()); in ParseArgumentShapes()
138 for (const auto& shape : llvm::enumerate(input_shapes_vector)) { in ParseArgumentShapes()
204 std::vector<llvm::Optional<std::vector<int>>> input_shapes_vector; in ParseXlaArguments() local
206 tensorflow::ParseNodeShapes(input_shapes_str, input_shapes_vector)); in ParseXlaArguments()
212 if (input_shapes_vector.empty()) in ParseXlaArguments()
213 input_shapes_vector.resize(dtypes_vector.size()); in ParseXlaArguments()
216 arg_kinds_vector.resize(input_shapes_vector.size(), in ParseXlaArguments()
219 if (input_shapes_vector.size() != dtypes_vector.size() || in ParseXlaArguments()
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| H A D | tf_mlir_translate.cc | 135 std::vector<llvm::Optional<std::vector<int>>> input_shapes_vector; in GraphdefToMlirTranslateFunction() local
141 TF_RETURN_IF_ERROR(ParseNodeShapes(input_shapes, input_shapes_vector)); in GraphdefToMlirTranslateFunction()
146 input_shapes_vector, output_array_vector, control_output_array_vector, in GraphdefToMlirTranslateFunction()
304 std::vector<llvm::Optional<std::vector<int>>> input_shapes_vector; in GraphdefToSplattedMlirTranslateFunction() local
310 TF_RETURN_IF_ERROR(ParseNodeShapes(input_shapes, input_shapes_vector)); in GraphdefToSplattedMlirTranslateFunction()
315 input_shapes_vector, output_array_vector, control_output_array_vector, in GraphdefToSplattedMlirTranslateFunction()
|