Home
last modified time | relevance | path

Searched refs:imported_dtype (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dmlir_roundtrip_flags.h32 DataType imported_dtype; member
Dmlir_roundtrip_flags.cc109 if (!DataType_Parse(used_node_dtypes[i], &info.imported_dtype)) { in ParseInputArrayInfo()
Dimport_model.cc465 DataType dtype = it->second.imported_dtype; in UpdateLegacyFedInputNode()
740 DataType dtype = array_info.imported_dtype; in AddNodesToShapeRefiner()
773 DataType dtype = index_it->second->second.imported_dtype; in AddNodesToShapeRefiner()
1303 array_info.imported_dtype = input_arg.type(); in ConvertLibFunction()
1367 DataType dtype = array_info.imported_dtype; in ConvertFeedsToPlaceholders()
2118 DataType dtype = node_info.imported_dtype; in InferLibFunctionType()
2381 DataType imported_dtype = node_info.imported_dtype; in InferMainFunctionType() local
2384 if (imported_dtype == DT_INVALID) { in InferMainFunctionType()
2385 imported_dtype = arg_node->output_type(0); in InferMainFunctionType()
2386 if (imported_dtype == DT_INVALID) { in InferMainFunctionType()
[all …]