Home
last modified time | relevance | path

Searched refs:ConvertToTfDataType (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/util/
H A Dattr_util.h48 DataType ConvertToTfDataType(tfrt::OpAttrType op_attr_type);
H A Dattr_util.cc222 DataType tf_dtype = ConvertToTfDataType(op_attr_type); in FillAttrValueMapUsingScalar()
290 DataType ConvertToTfDataType(tfrt::OpAttrType op_attr_type) { in ConvertToTfDataType() function
H A Dattr_util_test.cc133 EXPECT_EQ(ConvertToTfDataType(GetParam().op_attr_type), GetParam().dtype); in TEST_P()
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/kernel/
H A Dtfrt_op_kernel.cc64 *value = tfd::ConvertToTfDataType(attrtype); in GetAttr()
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/
H A Druntime_fallback_kernels.cc447 DataType tf_dtype = ConvertToTfDataType(op_attr_type); in PrepareAttributes()
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/eager/
H A Dc_api_tfrt.cc1276 *result = tensorflow::tfd::ConvertToTfDataType(optional_type.getValue()); in GetType()