Searched refs:input_shapes_vector (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | tf_xla_mlir_translate.cc | 121 std::vector<llvm::Optional<std::vector<int>>> input_shapes_vector; in ParseArgumentShapes() local
122 TF_RETURN_IF_ERROR(ParseNodeShapes(input_shapes_str, input_shapes_vector)); in ParseArgumentShapes()
123 arg_shapes.resize(input_shapes_vector.size()); in ParseArgumentShapes()
124 for (const auto& shape : llvm::enumerate(input_shapes_vector)) { in ParseArgumentShapes()
190 std::vector<llvm::Optional<std::vector<int>>> input_shapes_vector; in ParseXlaArguments() local
192 tensorflow::ParseNodeShapes(input_shapes_str, input_shapes_vector)); in ParseXlaArguments()
198 if (input_shapes_vector.empty()) in ParseXlaArguments()
199 input_shapes_vector.resize(dtypes_vector.size()); in ParseXlaArguments()
202 arg_kinds_vector.resize(input_shapes_vector.size(), in ParseXlaArguments()
205 if (input_shapes_vector.size() != dtypes_vector.size() || in ParseXlaArguments()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | tf_mlir_translate.cc | 132 std::vector<llvm::Optional<std::vector<int>>> input_shapes_vector; in GraphdefToMlirTranslateFunction() local
138 TF_RETURN_IF_ERROR(ParseNodeShapes(input_shapes, input_shapes_vector)); in GraphdefToMlirTranslateFunction()
143 input_shapes_vector, output_array_vector, control_output_array_vector, in GraphdefToMlirTranslateFunction()
292 std::vector<llvm::Optional<std::vector<int>>> input_shapes_vector; in GraphdefToSplattedMlirTranslateFunction() local
298 TF_RETURN_IF_ERROR(ParseNodeShapes(input_shapes, input_shapes_vector)); in GraphdefToSplattedMlirTranslateFunction()
303 input_shapes_vector, output_array_vector, control_output_array_vector, in GraphdefToSplattedMlirTranslateFunction()
|