| /external/federated-compute/fcp/aggregation/tensorflow/ |
| D | converters_test.cc | 62 EXPECT_EQ(*ConvertDataType(tf::DT_FLOAT), DT_FLOAT); in TEST()
63 EXPECT_EQ(*ConvertDataType(tf::DT_DOUBLE), DT_DOUBLE); in TEST()
64 EXPECT_EQ(*ConvertDataType(tf::DT_INT32), DT_INT32); in TEST()
65 EXPECT_EQ(*ConvertDataType(tf::DT_INT64), DT_INT64); in TEST()
66 EXPECT_EQ(*ConvertDataType(tf::DT_STRING), DT_STRING); in TEST()
70 EXPECT_THAT(ConvertDataType(tf::DT_VARIANT), IsCode(INVALID_ARGUMENT)); in TEST()
|
| D | converters.cc | 35 StatusOr<DataType> ConvertDataType(tf::DataType dtype) { in ConvertDataType() function
65 FCP_ASSIGN_OR_RETURN(DataType dtype, ConvertDataType(spec.dtype())); in ConvertTensorSpec()
130 FCP_ASSIGN_OR_RETURN(DataType dtype, ConvertDataType(tensor->dtype())); in ConvertTensor()
|
| D | converters.h | 37 StatusOr<DataType> ConvertDataType(::tensorflow::DataType dtype);
|
| D | checkpoint_reader.cc | 57 FCP_ASSIGN_OR_RETURN(DataType dtype, ConvertDataType(tf_dtype)); in Create()
|
| /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.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_type.h | 31 Status ConvertDataType(DataType dtype, mlir::Builder builder, mlir::Type* type);
|
| D | convert_tensor.cc | 121 TF_RETURN_IF_ERROR(ConvertDataType(input_dtype, *builder, &elt_type)); in ConvertTensor()
|
| D | compile_mlir_util.cc | 714 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()
310 TF_RETURN_IF_ERROR(ConvertDataType(dtype, builder, &mlir_type)); in SetAttrType()
383 ConvertDataType(output_arg.type(), builder, &type)); in Create()
426 TF_RETURN_IF_ERROR(ConvertDataType(output_arg.type(), builder, &type)); in Create()
593 tensorflow::ConvertDataType(arg_def.type(), builder, &expected_type)); in AddInput()
633 ConvertDataType(arg_def.type(), builder, &arg_def_type)); in AddInputList()
|
| /external/tensorflow/tensorflow/core/ir/importexport/ |
| D | convert_types.h | 29 tensorflow::Status ConvertDataType(tensorflow::DataType dtype, Builder& builder,
|
| D | convert_types.cc | 38 Status ConvertDataType(DataType dtype, Builder& builder, Type* type) { in ConvertDataType() function
199 TF_RETURN_IF_ERROR(ConvertDataType(dtype, *builder, &element_type)); in ConvertToMlirTensorType()
|
| D | convert_attributes.cc | 319 TF_RETURN_IF_ERROR(ConvertDataType(value.type(), builder, &type)); in ConvertNonFuncAttributeValue()
338 TF_RETURN_IF_ERROR(ConvertDataType(DataType(item), builder, &type)); in ConvertNonFuncAttributeValue()
462 TF_RETURN_IF_ERROR(ConvertDataType(handle.dtype(), builder, &dtype)); in ConvertHandleData()
|
| D | graphdef_import.cc | 556 TF_RETURN_IF_ERROR(ConvertDataType(it.value().type(), b_, &dtype)); in ConvertFunctionDef()
594 TF_RETURN_IF_ERROR(ConvertDataType(def.type(), b_, &dtype)); in ConvertFunctionDef()
729 TF_RETURN_IF_ERROR(ConvertDataType(arg_def.type(), b_, &dtype)); in ArgNumType()
896 TF_RETURN_IF_ERROR(ConvertDataType(tf_dtype, b_, &dtype)); in ConvertDataTypesToUnrankedTensorTypes()
|
| D | convert_tensor.cc | 118 TF_RETURN_IF_ERROR(ConvertDataType(input_dtype, builder, &elt_type)); in ConvertTensor()
|
| D | functiondef_import.cc | 231 TF_RETURN_IF_ERROR(ConvertDataType(arg.type(), builder, &input_type)); in ConvertArgDefAttributes()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/integration/ |
| D | tfr_decompose_ctx.cc | 136 TF_RETURN_IF_ERROR(ConvertDataType(ty, builder, &elt_ty)); in ExpandNode()
142 TF_RETURN_IF_ERROR(ConvertDataType(ty, builder, &elt_ty)); in ExpandNode()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | import_tensorflow.cc | 181 ArrayDataType ConvertDataType(tensorflow::DataType dtype) { in ConvertDataType() function
652 op->output_data_types.push_back(ConvertDataType(type)); in GetOutputTypesFromNodeDef()
741 op->output_data_types.push_back(ConvertDataType(output_types.type(i))); in ConvertUnsupportedOperator()
745 op->output_data_types.push_back(ConvertDataType(output_type)); in ConvertUnsupportedOperator()
1068 op->dtype = ConvertDataType(GetDataTypeAttr(node, "dtype")); in ConvertRandomUniform()
1585 array.data_type = ConvertDataType(GetDataTypeAttr(node, "dtype")); in ConvertPlaceholderOperator()
1617 op->src_data_type = ConvertDataType(tf_src_dtype); in ConvertCastOperator()
1618 op->dst_data_type = ConvertDataType(tf_dst_dtype); in ConvertCastOperator()
1724 op->output_data_type = ConvertDataType(output_type); in ConvertArgMinMaxOperator()
2061 op->dtype = ConvertDataType(dtype); in ConvertRangeOperator()
[all …]
|
| /external/tensorflow/tensorflow/security/advisory/ |
| D | tfsa-2022-140.md | 42 TF_RETURN_IF_ERROR(ConvertDataType(arg_def.type(), b_, &dtype));
|
| /external/tensorflow/tensorflow/core/ir/utils/ |
| D | shape_inference_utils.cc | 372 auto status = ConvertDataType(shape_n_type.dtype, b, &element_type); in InferReturnTypeComponentsForTFOp()
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | pywrap_tensor.cc | 162 int ConvertDataType(PyObject* obj, tensorflow::DataType* dst) { in ConvertDataType() function
516 ConvertDeviceName, &device_name, ConvertDataType, &dtype)) { in EagerTensor_init()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | import_model.cc | 1061 TF_RETURN_IF_ERROR(ConvertDataType(dtype, builder, &element_type)); in InferInputType()
1111 ConvertDataType(element_dtype->type(), builder, &etype)); in InferOutputType()
1156 TF_RETURN_IF_ERROR(ConvertDataType(dtype, builder, &element_type)); in InferOutputType()
1251 ::tensorflow::ConvertDataType(dtype, builder, &element_type)); in ConvertDataTypeAndShape()
1293 ::tensorflow::ConvertDataType(subtype.dtype, builder, &element_type)); in ConvertSubtypes()
2312 ::tensorflow::ConvertDataType(dtype, builder, &element_type)); in InferLibFunctionType()
2612 ConvertDataType(st.imported_dtype, builder, &st_data_type)); in InferMainFunctionType()
2627 ConvertDataType(imported_dtype, builder, &element_type)); in InferMainFunctionType()
|