Home
last modified time | relevance | path

Searched refs:TF_DataType (Results 1 – 25 of 62) sorted by relevance

123

/external/tensorflow/tensorflow/c/eager/
Ddlpack.cc74 DLDataType GetDlDataType(TF_DataType data_type, TF_Status* status) { in GetDlDataType()
79 case TF_DataType::TF_HALF: in GetDlDataType()
80 case TF_DataType::TF_FLOAT: in GetDlDataType()
81 case TF_DataType::TF_DOUBLE: in GetDlDataType()
84 case TF_DataType::TF_INT8: in GetDlDataType()
85 case TF_DataType::TF_INT16: in GetDlDataType()
86 case TF_DataType::TF_INT32: in GetDlDataType()
87 case TF_DataType::TF_INT64: in GetDlDataType()
90 case TF_DataType::TF_BOOL: in GetDlDataType()
91 case TF_DataType::TF_UINT8: in GetDlDataType()
[all …]
Dc_api_unified_experimental.h75 TF_DataType dtype, TF_Shape shape,
92 TF_DataType value, TF_Status* s);
Dc_api.h144 TF_CAPI_EXPORT extern TF_DataType TFE_TensorHandleDataType(TFE_TensorHandle* h);
314 TF_DataType value);
358 const TF_DataType* values,
/external/tensorflow/tensorflow/cc/experimental/base/tests/
Dtensor_types_test_util.h31 static constexpr TF_DataType kDType = TF_FLOAT;
36 static constexpr TF_DataType kDType = TF_DOUBLE;
41 static constexpr TF_DataType kDType = TF_INT32;
46 static constexpr TF_DataType kDType = TF_UINT8;
51 static constexpr TF_DataType kDType = TF_INT8;
56 static constexpr TF_DataType kDType = TF_INT64;
61 static constexpr TF_DataType kDType = TF_UINT16;
66 static constexpr TF_DataType kDType = TF_UINT32;
71 static constexpr TF_DataType kDType = TF_UINT64;
Dtensor_test.cc45 TF_DataType dtype = TypeParam::kDType; in TYPED_TEST()
69 TF_DataType dtype = TypeParam::kDType; in TYPED_TEST()
108 TF_DataType dtype = TypeParam::kDType; in TYPED_TEST()
Dtensorhandle_test.cc59 TF_DataType dtype = TypeParam::kDType; in TYPED_TEST()
96 TF_DataType dtype = TypeParam::kDType; in TYPED_TEST()
147 TF_DataType dtype = TypeParam::kDType; in TYPED_TEST()
/external/tensorflow/tensorflow/c/
Dtf_tensor.h89 TF_DataType, const int64_t* dims, int num_dims, void* data, size_t len,
102 TF_CAPI_EXPORT extern TF_Tensor* TF_AllocateTensor(TF_DataType,
114 TF_CAPI_EXPORT extern TF_DataType TF_TensorType(const TF_Tensor*);
157 TF_DataType type, TF_Tensor* to,
Dkernels.h111 const TF_DataType type, TF_Status* status);
181 TF_CAPI_EXPORT extern TF_DataType TF_ExpectedOutputDataType(
211 TF_OpKernelConstruction* ctx, const char* attr_name, TF_DataType* val,
269 TF_OpKernelConstruction* ctx, const char* attr_name, TF_DataType* vals,
339 int index, TF_DataType dtype,
360 TF_OpKernelContext* context, TF_DataType dtype, int64_t* dims, int num_dims,
Dc_api_experimental.h167 TF_CAPI_EXPORT extern TF_DataType TF_CheckpointReaderGetVariableDataType(
188 TF_DataType value);
191 const TF_DataType* values,
222 TF_DataType data_type, void* data, size_t len, TF_Status* status);
255 TF_DataType dtype;
277 TF_ShapeAndTypeList* shape_list, int index, TF_DataType dtype);
Dtf_datatype.h47 typedef enum TF_DataType { enum
72 } TF_DataType; typedef
77 TF_CAPI_EXPORT extern size_t TF_DataTypeSize(TF_DataType dt);
Dtf_tensor.cc70 TF_Tensor* CreateTensor(TF_ManagedBuffer* buf, TF_DataType dtype, in CreateTensor()
90 TF_Tensor* TF_AllocateTensor(TF_DataType dtype, const int64_t* dims, in TF_AllocateTensor()
100 TF_Tensor* TF_NewTensor(TF_DataType dtype, const int64_t* dims, int num_dims, in TF_NewTensor()
147 TF_DataType TF_TensorType(const TF_Tensor* t) { in TF_TensorType()
148 return static_cast<TF_DataType>(t->tensor->Type()); in TF_TensorType()
170 void TF_TensorBitcastFrom(const TF_Tensor* from, TF_DataType type, in TF_TensorBitcastFrom()
239 static TF_Tensor* EmptyTensor(TF_DataType dtype, in EmptyTensor()
267 return EmptyTensor(static_cast<TF_DataType>(src.dtype()), src.shape()); in TF_TensorFromTensor()
Dkernels.cc112 const TF_DataType type, in TF_KernelBuilder_TypeConstraint()
384 DEFINE_TF_GETATTR(Type, TF_DataType, tensorflow::DataType, "type", type)
450 TF_DataType TF_ExpectedOutputDataType(TF_OpKernelContext* ctx, int i) { in TF_ExpectedOutputDataType()
452 return static_cast<TF_DataType>(cc_ctx->expected_output_dtype(i)); in TF_ExpectedOutputDataType()
460 TF_DataType dtype, int64_t* dims, int num_dims, in TF_AllocateOutput()
513 TF_Tensor* TF_AllocateTemp(TF_OpKernelContext* context, TF_DataType dtype, in TF_AllocateTemp()
Dtf_datatype.cc20 size_t TF_DataTypeSize(TF_DataType dt) { in TF_DataTypeSize()
/external/tensorflow/tensorflow/cc/experimental/base/public/
Dtensor.h58 static Tensor FromBuffer(TF_DataType dtype, const std::vector<int64_t>& shape,
87 TF_DataType dtype() const;
135 inline TF_DataType Tensor::dtype() const { in dtype()
147 inline Tensor Tensor::FromBuffer(TF_DataType dtype, in FromBuffer()
/external/tensorflow/tensorflow/c/eager/parallel_device/
Dparallel_device_lib.h183 TF_DataType dtype() const { return dtype_; } in dtype()
188 absl::Span<const int64> shape, const TF_DataType dtype) in ParallelTensor()
194 std::vector<TensorHandlePtr> tensors, const TF_DataType dtype) in ParallelTensor()
206 const TF_DataType dtype_;
222 TF_DataType datatype_enum( in ScalarsFromSequence()
223 static_cast<TF_DataType>(DataTypeToEnum<DataType>().value)); in ScalarsFromSequence()
Dparallel_device_testlib.h41 Variable(TFE_TensorHandle* handle, TF_DataType type) in Variable()
46 static Variable* Create(TFE_Context* context, TF_DataType type,
73 TF_DataType type_;
147 static_cast<TF_DataType>(DataTypeToEnum<value_type>().value)); in ExpectScalarEq()
/external/tensorflow/tensorflow/c/kernels/ops/
Dbitcast.cc25 TF_ShapeHandle* shape, TF_DataType input_type, in ComputeNewShape()
26 TF_DataType output_type, TF_Status* status) { in ComputeNewShape()
81 TF_DataType input_type; in bitcast_shape_inference_fn()
82 TF_DataType output_type; in bitcast_shape_inference_fn()
/external/tensorflow/tensorflow/c/experimental/saved_model/internal/
Dtensor_spec.cc24 TF_DataType TF_TensorSpecDataType(const TF_TensorSpec* spec) { in TF_TensorSpecDataType()
25 return static_cast<TF_DataType>(tensorflow::unwrap(spec)->dtype()); in TF_TensorSpecDataType()
/external/tensorflow/tensorflow/lite/delegates/flex/
Dutil.cc32 tensor->type = GetTensorFlowLiteType(static_cast<TF_DataType>(src.dtype())); in CopyShapeAndType()
57 TF_DataType GetTensorFlowDataType(TfLiteType type) { in GetTensorFlowDataType()
96 TfLiteType GetTensorFlowLiteType(TF_DataType type) { in GetTensorFlowLiteType()
Dutil.h39 TF_DataType GetTensorFlowDataType(TfLiteType type);
42 TfLiteType GetTensorFlowLiteType(TF_DataType);
/external/tensorflow/tensorflow/java/src/main/native/
Dtensor_jni.cc38 size_t elemByteSize(TF_DataType dtype) { in elemByteSize()
69 void writeScalar(JNIEnv* env, jobject src, TF_DataType dtype, void* dst, in writeScalar()
114 size_t write1DArray(JNIEnv* env, jarray array, TF_DataType dtype, void* dst, in write1DArray()
152 size_t read1DArray(JNIEnv* env, TF_DataType dtype, const void* src, in read1DArray()
185 size_t writeNDArray(JNIEnv* env, jarray src, TF_DataType dtype, int dims_left, in writeNDArray()
204 size_t readNDArray(JNIEnv* env, TF_DataType dtype, const char* src, in readNDArray()
311 TF_Tensor* t = TF_AllocateTensor(static_cast<TF_DataType>(dtype), dims_copy, in Java_org_tensorflow_Tensor_allocate()
446 static_assert(sizeof(jint) >= sizeof(TF_DataType), in Java_org_tensorflow_Tensor_dtype()
477 TF_DataType dtype = TF_TensorType(t); in Java_org_tensorflow_Tensor_setValue()
536 TF_DataType dtype = TF_TensorType(t); in Java_org_tensorflow_Tensor_readNDArray()
/external/tensorflow/tensorflow/python/lib/core/
Dndarray_tensor_bridge.cc110 Status TF_DataType_to_PyArray_TYPE(TF_DataType tf_datatype, in TF_DataType_to_PyArray_TYPE()
198 TF_DataType_to_PyArray_TYPE(static_cast<TF_DataType>(dtype), &type_num); in ArrayFromMemory()
/external/tensorflow/tensorflow/c/kernels/
Dbitcast_op.cc46 TF_DataType input_data_type;
47 TF_DataType output_data_type;
/external/tensorflow/tensorflow/c/experimental/saved_model/public/
Dtensor_spec.h33 TF_CAPI_EXPORT extern TF_DataType TF_TensorSpecDataType(
/external/tensorflow/tensorflow/python/client/
Dtf_session_helper.cc624 const std::vector<int>& ranks, const std::vector<TF_DataType>& types, in TF_GraphSetOutputHandleShapesAndTypes_wrapper()
636 TF_DataType dtype, TF_Output* output) { in CreatePlaceholder()
665 TF_DataType tf_datatype = static_cast<TF_DataType>(PyLong_AsLong(dtype)); in TF_CreatePlaceholders()
667 TF_DataType tf_datatype = static_cast<TF_DataType>(PyInt_AsLong(dtype)); in TF_CreatePlaceholders()

123