Home
last modified time | relevance | path

Searched refs:DType (Results 1 – 25 of 136) sorted by relevance

123456

/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/python_binding/
Dconversion_utils.cc25 using ::tfrt::DType;
30 const char* ToPythonStructFormat(DType dtype_kind) { in ToPythonStructFormat()
34 case DType::Invalid: in ToPythonStructFormat()
36 case DType::Unsupported: in ToPythonStructFormat()
38 case DType::UI8: in ToPythonStructFormat()
40 case DType::UI16: in ToPythonStructFormat()
42 case DType::UI32: in ToPythonStructFormat()
44 case DType::UI64: in ToPythonStructFormat()
46 case DType::I1: in ToPythonStructFormat()
48 case DType::I8: in ToPythonStructFormat()
[all …]
Dtf_cpurt_executor.cc44 using ::tfrt::DType;
92 static const char* ToPythonStructFormat(DType dtype_kind) { in ToPythonStructFormat()
96 case DType::Invalid: in ToPythonStructFormat()
98 case DType::Unsupported: in ToPythonStructFormat()
100 case DType::UI8: in ToPythonStructFormat()
102 case DType::UI16: in ToPythonStructFormat()
104 case DType::UI32: in ToPythonStructFormat()
106 case DType::UI64: in ToPythonStructFormat()
108 case DType::I1: in ToPythonStructFormat()
110 case DType::I8: in ToPythonStructFormat()
[all …]
/external/tensorflow/tensorflow/core/runtime_fallback/util/
Dtype_util_test.cc27 tfrt::DType tfrt_dtype;
37 TfrtDTypeAndTfDataType{tfrt::DType::UI8, DataType::DT_UINT8},
38 TfrtDTypeAndTfDataType{tfrt::DType::UI16, DataType::DT_UINT16},
39 TfrtDTypeAndTfDataType{tfrt::DType::UI32, DataType::DT_UINT32},
40 TfrtDTypeAndTfDataType{tfrt::DType::UI64, DataType::DT_UINT64},
41 TfrtDTypeAndTfDataType{tfrt::DType::I1, DataType::DT_BOOL},
42 TfrtDTypeAndTfDataType{tfrt::DType::I8, DataType::DT_INT8},
43 TfrtDTypeAndTfDataType{tfrt::DType::I16, DataType::DT_INT16},
44 TfrtDTypeAndTfDataType{tfrt::DType::I32, DataType::DT_INT32},
45 TfrtDTypeAndTfDataType{tfrt::DType::I64, DataType::DT_INT64},
[all …]
Dtype_util.h27 inline DataType GetTfDataType(tfrt::DType dtype) { in GetTfDataType()
29 case tfrt::DType::Invalid: in GetTfDataType()
30 case tfrt::DType::Unsupported: in GetTfDataType()
31 case tfrt::DType::Resource: in GetTfDataType()
35 case tfrt::DType::TFRT_ENUM: \ in GetTfDataType()
41 inline tfrt::DType GetTfrtDtype(DataType dtype) { in GetTfrtDtype()
44 return tfrt::DType(tfrt::DType::Unsupported); in GetTfrtDtype()
46 return tfrt::DType(); in GetTfrtDtype()
48 return tfrt::DType(tfrt::DType::Resource); in GetTfrtDtype()
51 return tfrt::DType(tfrt::DType::TFRT_ENUM); in GetTfrtDtype()
Dattr_util.cc122 llvm::makeArrayRef(static_cast<const tfrt::DType*>(op_attr.GetData()), in FillAttrValueMapUsingArray()
158 GetDataType(attr_base.type()) == tfrt::DType::String) { in FillAttrValueMapUsingAggregate()
316 DataType ConvertBefAttrTypeToTfDataType(tfrt::DType attr_type) { in ConvertBefAttrTypeToTfDataType()
318 case tfrt::DType::I1: in ConvertBefAttrTypeToTfDataType()
320 case tfrt::DType::I8: in ConvertBefAttrTypeToTfDataType()
322 case tfrt::DType::I16: in ConvertBefAttrTypeToTfDataType()
324 case tfrt::DType::I32: in ConvertBefAttrTypeToTfDataType()
326 case tfrt::DType::I64: in ConvertBefAttrTypeToTfDataType()
328 case tfrt::DType::UI8: in ConvertBefAttrTypeToTfDataType()
330 case tfrt::DType::UI16: in ConvertBefAttrTypeToTfDataType()
[all …]
Dattr_util_test.cc146 tfrt::DType tfrt_dtype;
156 TfrtDTypeAndTensorflowDType{tfrt::DType::I1, DataType::DT_BOOL},
157 TfrtDTypeAndTensorflowDType{tfrt::DType::UI8, DataType::DT_UINT8},
158 TfrtDTypeAndTensorflowDType{tfrt::DType::I8, DataType::DT_INT8},
159 TfrtDTypeAndTensorflowDType{tfrt::DType::I16, DataType::DT_INT16},
160 TfrtDTypeAndTensorflowDType{tfrt::DType::UI16, DataType::DT_UINT16},
161 TfrtDTypeAndTensorflowDType{tfrt::DType::I32, DataType::DT_INT32},
162 TfrtDTypeAndTensorflowDType{tfrt::DType::UI32, DataType::DT_UINT32},
163 TfrtDTypeAndTensorflowDType{tfrt::DType::I64, DataType::DT_INT64},
164 TfrtDTypeAndTensorflowDType{tfrt::DType::UI64, DataType::DT_UINT64},
[all …]
/external/tensorflow/tensorflow/lite/kernels/shim/
Dtensor_view.h73 template <typename DType, int RANK>
77 : data_(t->Data<DType>()), shape_(t->Shape()) { in Tensor()
83 : data_(t->Data<DType>()), shape_(t->Shape()) { in Tensor()
90 inline DType &operator()(IndexTypes... indices) { in operator()
97 inline const DType &operator()(IndexTypes... indices) const { in operator()
103 typename absl::Span<DType>::pointer Ptr() { return data_.data(); } in Ptr()
104 constexpr typename absl::Span<DType>::const_pointer Ptr() const { in Ptr()
140 absl::Span<DType> data_;
172 template <typename DType>
173 absl::Span<DType> &Data() { in Data()
[all …]
Dtf_tensor_view.h45 template <typename DType>
46 TfTensorView(const ::tensorflow::Tensor *wrapped_tensor, const DType &dtype);
84 template <typename DType>
86 const DType &dtype)
Dtflite_tensor_view.h47 template <typename DType>
48 TfLiteTensorView(::TfLiteTensor *wrapped_tensor, const DType &dtype) in TfLiteTensorView()
60 template <typename DType>
61 TfLiteTensorView(const ::TfLiteTensor *wrapped_tensor, const DType &dtype) in TfLiteTensorView()
/external/tensorflow/tensorflow/compiler/xla/
Dtfrt_utils_test.cc24 using ::tfrt::DType;
27 DType dtype;
33 {DType::UI8, PrimitiveType::U8},
34 {DType::UI16, PrimitiveType::U16},
35 {DType::UI32, PrimitiveType::U32},
36 {DType::UI64, PrimitiveType::U64},
37 {DType::I1, PrimitiveType::PRED},
38 {DType::I8, PrimitiveType::S8},
39 {DType::I16, PrimitiveType::S16},
40 {DType::I32, PrimitiveType::S32},
[all …]
Dtfrt_utils.cc21 PrimitiveType TfrtToPrimitiveType(tfrt::DType dtype) { in TfrtToPrimitiveType()
24 case tfrt::DType::UI8: in TfrtToPrimitiveType()
26 case tfrt::DType::UI16: in TfrtToPrimitiveType()
28 case tfrt::DType::UI32: in TfrtToPrimitiveType()
30 case tfrt::DType::UI64: in TfrtToPrimitiveType()
34 case tfrt::DType::I1: in TfrtToPrimitiveType()
36 case tfrt::DType::I8: in TfrtToPrimitiveType()
38 case tfrt::DType::I16: in TfrtToPrimitiveType()
40 case tfrt::DType::I32: in TfrtToPrimitiveType()
42 case tfrt::DType::I64: in TfrtToPrimitiveType()
[all …]
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.dtypes.pbtxt4 name: "DType"
13 mtype: "<class \'tensorflow.python.framework.dtypes.DType\'>"
17 mtype: "<class \'tensorflow.python.framework.dtypes.DType\'>"
21 mtype: "<class \'tensorflow.python.framework.dtypes.DType\'>"
25 mtype: "<class \'tensorflow.python.framework.dtypes.DType\'>"
29 mtype: "<class \'tensorflow.python.framework.dtypes.DType\'>"
33 mtype: "<class \'tensorflow.python.framework.dtypes.DType\'>"
37 mtype: "<class \'tensorflow.python.framework.dtypes.DType\'>"
41 mtype: "<class \'tensorflow.python.framework.dtypes.DType\'>"
45 mtype: "<class \'tensorflow.python.framework.dtypes.DType\'>"
[all …]
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.dtypes.pbtxt4 name: "DType"
13 mtype: "<class \'tensorflow.python.framework.dtypes.DType\'>"
17 mtype: "<class \'tensorflow.python.framework.dtypes.DType\'>"
21 mtype: "<class \'tensorflow.python.framework.dtypes.DType\'>"
25 mtype: "<class \'tensorflow.python.framework.dtypes.DType\'>"
29 mtype: "<class \'tensorflow.python.framework.dtypes.DType\'>"
33 mtype: "<class \'tensorflow.python.framework.dtypes.DType\'>"
37 mtype: "<class \'tensorflow.python.framework.dtypes.DType\'>"
41 mtype: "<class \'tensorflow.python.framework.dtypes.DType\'>"
45 mtype: "<class \'tensorflow.python.framework.dtypes.DType\'>"
[all …]
Dtensorflow.lite.constants.pbtxt5 mtype: "<class \'tensorflow.python.framework.dtypes.DType\'>"
9 mtype: "<class \'tensorflow.python.framework.dtypes.DType\'>"
17 mtype: "<class \'tensorflow.python.framework.dtypes.DType\'>"
21 mtype: "<class \'tensorflow.python.framework.dtypes.DType\'>"
25 mtype: "<class \'tensorflow.python.framework.dtypes.DType\'>"
29 mtype: "<class \'tensorflow.python.framework.dtypes.DType\'>"
33 mtype: "<class \'tensorflow.python.framework.dtypes.DType\'>"
37 mtype: "<class \'tensorflow.python.framework.dtypes.DType\'>"
/external/tensorflow/tensorflow/python/framework/
Ddtypes.py37 class DTypeMeta(type(_dtypes.DType), abc.ABCMeta):
42 class DType( class
43 _dtypes.DType,
209 return DType(proto.datatype)
220 if type(other) != DType: # pylint: disable=unidiomatic-typecheck
236 __hash__ = _dtypes.DType.__hash__
241 trace_type.register_serializable(DType)
260 resource = DType(types_pb2.DT_RESOURCE)
266 variant = DType(types_pb2.DT_VARIANT)
272 uint8 = DType(types_pb2.DT_UINT8)
[all …]
/external/tensorflow/tensorflow/compiler/xla/mlir/transforms/runtime/
Dtype_converter.cc30 using tfrt::DType;
78 /*static*/ Expected<DType> TypeConverter::ConvertElementType(mlir::Type type) { in ConvertElementType()
79 if (type.isF32()) return DType::F32; in ConvertElementType()
80 if (type.isF64()) return DType::F64; in ConvertElementType()
81 if (type.isUnsignedInteger(8)) return DType::UI8; in ConvertElementType()
82 if (type.isUnsignedInteger(16)) return DType::UI16; in ConvertElementType()
83 if (type.isUnsignedInteger(32)) return DType::UI32; in ConvertElementType()
84 if (type.isUnsignedInteger(64)) return DType::UI64; in ConvertElementType()
85 if (type.isInteger(1)) return DType::I1; in ConvertElementType()
86 if (type.isInteger(8)) return DType::I8; in ConvertElementType()
[all …]
/external/armnn/src/backends/tosaCommon/
DTosaLayerSupportRules.hpp27 m_Res = std::any_of(c.begin(), c.end(), [&tensor](DType dt) in TosaTypeAnyOf()
53 explicit TosaContainerContainsTwoTypes(std::tuple<DType, DType>& check, in TosaContainerContainsTwoTypes()
54 const std::vector<std::tuple<DType, DType>>& c) in TosaContainerContainsTwoTypes()
71 explicit TosaContainerContainsThreeTypes(std::tuple<DType, DType, DType>& check, in TosaContainerContainsThreeTypes()
72 const std::vector<std::tuple<DType, DType, DType>>& c) in TosaContainerContainsThreeTypes()
/external/tensorflow/tensorflow/compiler/xla/runtime/
Dsymbolic_shape_test.cc33 using tfrt::DType;
39 llvm::SmallVector<DType> dtypes, in GetFunctionType()
63 return MemrefDesc(DType::F32, nullptr, 0, shape, shape /* fake strides */); in GetFakeMemref()
83 auto dtypes = {DType::F32, DType::I32, DType::I1}; in TEST()
146 auto dtypes = {DType::F32, DType::I32, DType::I1}; in TEST()
208 auto dtypes = {DType::F32, DType::I32, DType::I1}; in TEST()
282 auto dtypes = {DType::F32, DType::I32}; in TEST()
306 auto dtypes = {DType::F32, DType::I32}; in TEST()
345 auto dtypes = {DType::F32, DType::I32}; in TEST()
371 auto dtypes = {DType::F32}; in TEST()
[all …]
Darguments.cc39 using tfrt::DType;
80 static bool AreCompatibleTypes(DType type1, DType type2) { in AreCompatibleTypes()
81 auto compatible = [&](DType fromType, DType toType) { in AreCompatibleTypes()
86 if (compatible(DType::I1, DType::I8)) return true; in AreCompatibleTypes()
89 if (compatible(DType::I8, DType::UI8) || in AreCompatibleTypes()
90 compatible(DType::I16, DType::UI16) || in AreCompatibleTypes()
91 compatible(DType::I32, DType::UI32) || in AreCompatibleTypes()
92 compatible(DType::I64, DType::UI64)) in AreCompatibleTypes()
98 static Error VerifyMemrefArgument(DType element_type, in VerifyMemrefArgument()
110 auto print_shaped = [&](Optional<ArrayRef<int64_t>> dims, DType dtype) { in VerifyMemrefArgument()
[all …]
Dtypes.h139 RankedTensorType(llvm::ArrayRef<int64_t> sizes, tfrt::DType element_type) in RankedTensorType()
144 tfrt::DType element_type() const { return element_type_; } in element_type()
150 tfrt::DType element_type_;
161 explicit UnrankedTensorType(tfrt::DType element_type) in UnrankedTensorType()
164 tfrt::DType element_type() const { return element_type_; } in element_type()
169 tfrt::DType element_type_;
183 MemrefType(llvm::ArrayRef<int64_t> sizes, tfrt::DType element_type) in MemrefType()
188 tfrt::DType element_type() const { return element_type_; } in element_type()
197 tfrt::DType element_type_;
208 explicit UnrankedMemrefType(tfrt::DType element_type) in UnrankedMemrefType()
[all …]
/external/tensorflow/tensorflow/core/tfrt/eager/
Dc_api_tfrt.cc111 TensorMetadata CreateMetadata(DType dtype, absl::Span<const Index> dim_sizes) { in CreateMetadata()
113 DType(dtype), in CreateMetadata()
118 tensorflow::DataType ConvertDType(DType kind) { in ConvertDType()
120 case DType::UI8: in ConvertDType()
122 case DType::UI16: in ConvertDType()
124 case DType::UI32: in ConvertDType()
126 case DType::UI64: in ConvertDType()
128 case DType::I8: in ConvertDType()
130 case DType::I16: in ConvertDType()
132 case DType::I32: in ConvertDType()
[all …]
/external/tensorflow/tensorflow/go/
Dsignature_test.go93 DType: Int8,
98 DType: Float,
105 DType: String,
110 DType: Bool,
158 DType: Int8,
176 if a.DType != b.DType {
177 diff += fmt.Sprintf("DType: %v -> %v\n", a.DType, b.DType)
/external/armnn/src/backends/tosaCommon/operatorMappings/
DTransposeConv2dOperator.cpp43 DType inputDType0 = ArmNNToDType(inputs[0]->GetDataType()); in ConvertTransposeConv2dToTosaOperator()
56 DType inputDType1 = ArmNNToDType(inputs[1]->GetDataType()); in ConvertTransposeConv2dToTosaOperator()
66 … DType inputDType1 = ArmNNToDType(transposeConv2dLayer->m_Weight->GetTensorInfo().GetDataType()); in ConvertTransposeConv2dToTosaOperator()
80 DType inputDType2 = ArmNNToDType(inputs[2]->GetDataType()); in ConvertTransposeConv2dToTosaOperator()
90 … DType inputDType2 = ArmNNToDType(transposeConv2dLayer->m_Bias->GetTensorInfo().GetDataType()); in ConvertTransposeConv2dToTosaOperator()
114 DType outputDType0 = ArmNNToDType(outputs[0]->GetDataType()); in ConvertTransposeConv2dToTosaOperator()
/external/tensorflow/tensorflow/core/runtime_fallback/runtime/
Druntime_fallback_batch_kernels.cc84 case tfrt::DType::I1: in ExecuteFallbackOp()
87 case tfrt::DType::I32: in ExecuteFallbackOp()
90 case tfrt::DType::I64: in ExecuteFallbackOp()
93 case tfrt::DType::F32: in ExecuteFallbackOp()
96 case tfrt::DType::F64: in ExecuteFallbackOp()
99 case tfrt::DType::String: in ExecuteFallbackOp()
109 tfrt::DType type = type_attr.GetValue(); in ExecuteFallbackOp()
/external/tensorflow/tensorflow/core/tfrt/utils/
Dtensor_util.cc87 StatusOr<DType> ConvertTFDTypeToTFRTDType(tensorflow::DataType dtype) { in ConvertTFDTypeToTFRTDType()
91 return DType(DType::TFRT_DTYPE); in ConvertTFDTypeToTFRTDType()
99 StatusOr<tensorflow::DataType> ConvertTFRTDTypeToTFDType(DType dtype) { in ConvertTFRTDTypeToTFDType()
102 case DType::TFRT_DTYPE: \ in ConvertTFRTDTypeToTFDType()
151 if (dtype == DType::String) { in CreateTensorHandleFromTFTensor()

123456