Searched refs:ConvertDataType (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | convert_attr.cc | 44 TF_RETURN_IF_ERROR(ConvertDataType(value.type(), *builder, &type)); in ConvertNonFuncAttributeValue()
63 TF_RETURN_IF_ERROR(ConvertDataType(DataType(item), *builder, &type)); in ConvertNonFuncAttributeValue()
|
| D | convert_type.h | 31 Status ConvertDataType(DataType dtype, mlir::Builder builder, mlir::Type* type);
|
| D | convert_type.cc | 34 Status ConvertDataType(DataType dtype, Builder builder, Type* type) { in ConvertDataType() function
187 TF_RETURN_IF_ERROR(ConvertDataType(dtype, *builder, &element_type)); in ConvertToMlirTensorType()
|
| D | convert_tensor.cc | 126 TF_RETURN_IF_ERROR(ConvertDataType(input_dtype, *builder, &elt_type)); in ConvertTensor()
|
| D | shape_inference_utils.cc | 321 tensorflow::ConvertDataType(shape_n_type.dtype, b, &element_type); in InferReturnTypeComponentsForTFOp()
|
| D | compile_mlir_util.cc | 537 TF_RETURN_IF_ERROR(ConvertDataType(xla_arg.type, builder, &element_type)); in RewriteWithArgs()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/c/ |
| D | c_api_unified_experimental_mlir.cc | 88 Status s = tensorflow::ConvertDataType(dtype, builder, type); in ConvertDataTypeToTensor()
309 TF_RETURN_IF_ERROR(ConvertDataType(dtype, builder, &mlir_type)); in SetAttrType()
382 ConvertDataType(output_arg.type(), builder, &type)); in Create()
425 TF_RETURN_IF_ERROR(ConvertDataType(output_arg.type(), builder, &type)); in Create()
590 tensorflow::ConvertDataType(arg_def.type(), builder, &expected_type)); in AddInput()
630 ConvertDataType(arg_def.type(), builder, &arg_def_type)); in AddInputList()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/integration/ |
| D | tfr_decompose_ctx.cc | 134 TF_RETURN_IF_ERROR(ConvertDataType(ty, builder, &elt_ty)); in ExpandNode()
140 TF_RETURN_IF_ERROR(ConvertDataType(ty, builder, &elt_ty)); in ExpandNode()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | import_tensorflow.cc | 180 ArrayDataType ConvertDataType(tensorflow::DataType dtype) { in ConvertDataType() function
645 op->output_data_types.push_back(ConvertDataType(type)); in GetOutputTypesFromNodeDef()
734 op->output_data_types.push_back(ConvertDataType(output_types.type(i))); in ConvertUnsupportedOperator()
738 op->output_data_types.push_back(ConvertDataType(output_type)); in ConvertUnsupportedOperator()
1061 op->dtype = ConvertDataType(GetDataTypeAttr(node, "dtype")); in ConvertRandomUniform()
1578 array.data_type = ConvertDataType(GetDataTypeAttr(node, "dtype")); in ConvertPlaceholderOperator()
1610 op->src_data_type = ConvertDataType(tf_src_dtype); in ConvertCastOperator()
1611 op->dst_data_type = ConvertDataType(tf_dst_dtype); in ConvertCastOperator()
1717 op->output_data_type = ConvertDataType(output_type); in ConvertArgMinMaxOperator()
2054 op->dtype = ConvertDataType(dtype); in ConvertRangeOperator()
[all …]
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | pywrap_tensor.cc | 156 int ConvertDataType(PyObject* obj, tensorflow::DataType* dst) { in ConvertDataType() function
458 ConvertDeviceName, &device_name, ConvertDataType, &dtype)) { in EagerTensor_init()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | import_model.cc | 951 TF_RETURN_IF_ERROR(ConvertDataType(dtype, builder, &element_type)); in InferInputType()
981 ConvertDataType(element_dtype->type(), builder, &etype)); in InferOutputType()
1026 TF_RETURN_IF_ERROR(ConvertDataType(dtype, builder, &element_type)); in InferOutputType()
1104 ::tensorflow::ConvertDataType(dtype, builder, &element_type)); in ConvertDataTypeAndShape()
1146 ::tensorflow::ConvertDataType(subtype.dtype, builder, &element_type)); in ConvertSubtypes()
2131 ::tensorflow::ConvertDataType(dtype, builder, &element_type)); in InferLibFunctionType()
2391 ::tensorflow::ConvertDataType(imported_dtype, builder, &element_type)); in InferMainFunctionType()
|