Searched refs:ConvertToDataType (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | convert_type.h | 38 Status ConvertToDataType(mlir::Type type, DataType* dtype);
|
| D | compile_mlir_util.cc | 103 TF_RETURN_IF_ERROR(ConvertToDataType(func_type.getInput(i), &dtype)); in GetXlaInputShapes()
177 TF_RETURN_IF_ERROR(ConvertToDataType(tensor_type, &resource_update.type)); in GetOutputInfo()
184 TF_RETURN_IF_ERROR(ConvertToDataType(tensor_type, &out_desc.type)); in GetOutputInfo()
|
| D | convert_type.cc | 152 Status ConvertToDataType(Type type, DataType* dtype) { in ConvertToDataType() function
|
| D | convert_tensor.cc | 313 TF_RETURN_IF_ERROR(ConvertToDataType(type, &output_dtype)); in ConvertToTensorProto()
|
| D | shape_inference_utils.cc | 113 tensorflow::ConvertToDataType(subtype.getElementType(), &dtype); in GetSubtypesHelper()
|
| D | export_utils.cc | 165 TF_RETURN_IF_ERROR(ConvertToDataType(type, &dtype)); in ConvertAttribute()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | type_to_shape.cc | 95 TF_RETURN_IF_ERROR(tensorflow::ConvertToDataType(type, &dtype)); in TypeToShape()
|
| D | mlir_hlo_to_hlo.cc | 1458 status = tensorflow::ConvertToDataType(arg.getType(), &dtype); in SetEntryTupleShapesAndLeafReplication()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | export_graphdef.cc | 228 TF_RETURN_IF_ERROR(ConvertToDataType( in GetArgumentNode()
264 TF_RETURN_IF_ERROR(ConvertToDataType( in GetReturnNode()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/c/ |
| D | c_api_unified_experimental_mlir.cc | 100 Status s = ConvertToDataType(value_.getType(), &type); in DataType()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_with_tf2xla.cc | 565 auto status = tensorflow::ConvertToDataType(operand.getType(), &dtype); in GetExprForOperand()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_rewrite_pass.cc | 239 tensorflow::ConvertToDataType(operand_type, &dtype); in SetMetadataProtoArgs()
|