Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/c/
Dc_api_experimental.h165 TF_Session* session, int tensor_id, TF_DataType inputType,
169 TFE_Context* ctx, int tensor_id, TF_DataType inputType, TF_Status* status);
225 TF_DataType value);
228 const TF_DataType* values,
259 TF_DataType dtype, void* scalar, size_t len);
272 TF_Output t, TF_DataType data_type);
Dc_api.h99 typedef enum TF_DataType { enum
124 } TF_DataType; typedef
129 TF_CAPI_EXPORT extern size_t TF_DataTypeSize(TF_DataType dt);
235 TF_DataType, const int64_t* dims, int num_dims, void* data, size_t len,
248 TF_CAPI_EXPORT extern TF_Tensor* TF_AllocateTensor(TF_DataType,
260 TF_CAPI_EXPORT extern TF_DataType TF_TensorType(const TF_Tensor*);
303 TF_DataType type, TF_Tensor* to,
547 TF_DataType value);
550 const TF_DataType* values,
630 TF_CAPI_EXPORT extern TF_DataType TF_OperationOutputType(TF_Output oper_out);
[all …]
Dkernels.h125 TF_CAPI_EXPORT extern TF_DataType TF_ExpectedOutputDataType(
137 TF_OpKernelConstruction* ctx, const char* attr_name, TF_DataType* val,
Dc_api_internal.h60 TF_DataType dtype;
200 static Tensor MakeTensor(TF_DataType type, const TensorShape& shape, in MakeTensor()
225 const TF_DataType* types,
Dkernels.cc190 DEFINE_TF_GETATTR(Type, TF_DataType, tensorflow::DataType) in DEFINE_TF_GETATTR() argument
192 TF_DataType TF_ExpectedOutputDataType(TF_OpKernelContext* ctx, int i) { in DEFINE_TF_GETATTR()
194 return static_cast<TF_DataType>(cc_ctx->expected_output_dtype(i)); in DEFINE_TF_GETATTR()
Dc_api_test.cc293 EXPECT_EQ(TF_FLOAT, static_cast<TF_DataType>(tensorflow::DT_FLOAT)); in TEST()
294 EXPECT_EQ(TF_DOUBLE, static_cast<TF_DataType>(tensorflow::DT_DOUBLE)); in TEST()
295 EXPECT_EQ(TF_INT32, static_cast<TF_DataType>(tensorflow::DT_INT32)); in TEST()
296 EXPECT_EQ(TF_UINT8, static_cast<TF_DataType>(tensorflow::DT_UINT8)); in TEST()
297 EXPECT_EQ(TF_INT16, static_cast<TF_DataType>(tensorflow::DT_INT16)); in TEST()
298 EXPECT_EQ(TF_INT8, static_cast<TF_DataType>(tensorflow::DT_INT8)); in TEST()
299 EXPECT_EQ(TF_STRING, static_cast<TF_DataType>(tensorflow::DT_STRING)); in TEST()
300 EXPECT_EQ(TF_COMPLEX64, static_cast<TF_DataType>(tensorflow::DT_COMPLEX64)); in TEST()
302 EXPECT_EQ(TF_INT64, static_cast<TF_DataType>(tensorflow::DT_INT64)); in TEST()
303 EXPECT_EQ(TF_BOOL, static_cast<TF_DataType>(tensorflow::DT_BOOL)); in TEST()
[all …]
Dc_test_util.h54 TF_DataType dtype = TF_INT32,
101 TF_Operation* RandomUniform(TF_Operation* shape, TF_DataType dtype,
Dc_api.cc112 size_t TF_DataTypeSize(TF_DataType dt) { in TF_DataTypeSize()
197 TF_Tensor* TF_AllocateTensor(TF_DataType dtype, const int64_t* dims, in TF_AllocateTensor()
204 TF_Tensor* TF_NewTensor(TF_DataType dtype, const int64_t* dims, int num_dims, in TF_NewTensor()
257 TF_DataType TF_TensorType(const TF_Tensor* t) { return t->dtype; } in TF_TensorType()
296 void TF_TensorBitcastFrom(const TF_Tensor* from, TF_DataType type, in TF_TensorBitcastFrom()
545 static TF_Tensor* EmptyTensor(TF_DataType dtype, const TensorShape& shape) { in EmptyTensor()
571 return EmptyTensor(static_cast<TF_DataType>(src.dtype()), src.shape()); in TF_TensorFromTensor()
592 return new TF_Tensor{static_cast<TF_DataType>(src.dtype()), src.shape(), in TF_TensorFromTensor()
725 const TF_DataType* types, in TF_GraphSetOutputHandleShapesAndTypes()
882 EmptyTensor(static_cast<TF_DataType>(src.dtype()), src.shape()); in TF_Run_Helper()
[all …]
Dc_test.c31 TF_DataType type; in create()
Dc_api_experimental_test.cc409 TF_DataType dtype; in TEST_F()
434 TF_DataType dtype; in TEST_F()
457 TF_DataType types[kNumInputs] = {TF_FLOAT, TF_FLOAT, TF_FLOAT}; in TEST_F()
/external/tensorflow/tensorflow/python/eager/
Dpywrap_tensor.h29 bool IsCompatible(int desired_dtype, TF_DataType returned_dtype);
35 TF_DataType src_type_enum,
36 TF_DataType dst_type_enum, TF_Status* out_status);
Dpywrap_tensor.cc129 PyObject* PyIntFromDataType(TF_DataType l) { in PyIntFromDataType()
147 bool IsCompatible(int desired_dtype, TF_DataType returned_dtype) { in IsCompatible()
177 TF_DataType src_type_enum, in EagerCast()
178 TF_DataType dst_type_enum, TF_Status* out_status) { in EagerCast()
235 static_cast<TF_DataType>(desired_dtype), &desired_np_dtype) in ConvertToEagerTensor()
421 TF_DataType handle_dtype = TFE_TensorHandleDataType(handle.get()); in EagerTensor_init()
427 static_cast<TF_DataType>(desired_dtype), self->status)); in EagerTensor_init()
Dpywrap_tfe_src.cc312 reinterpret_cast<const TF_DataType*>(values.get()), in SetOpAttrList()
477 reinterpret_cast<const TF_DataType*>(values.get()), in SetOpAttrListDefault()
552 TFE_OpSetAttrType(op, key, static_cast<TF_DataType>(value)); in SetOpAttrScalar()
2182 const TF_DataType& desired_dtype,
2185 TF_DataType input_dtype = TFE_TensorHandleDataType(handle->get());
2186 TF_DataType output_dtype = input_dtype;
2191 static_cast<TF_DataType>(desired_dtype), status));
2226 TFE_OpSetAttrType(op, "dtype", static_cast<TF_DataType>(value));
2285 const std::function<void(const TF_DataType& dtype)>& dtype_setter,
2322 TF_DataType output_dtype = TFE_TensorHandleDataType(handle.get());
[all …]
/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()
82 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()
351 TF_Tensor* t = TF_AllocateTensor(static_cast<TF_DataType>(dtype), dims_copy, in Java_org_tensorflow_Tensor_allocate()
497 static_assert(sizeof(jint) >= sizeof(TF_DataType), in Java_org_tensorflow_Tensor_dtype()
528 TF_DataType dtype = TF_TensorType(t); in Java_org_tensorflow_Tensor_setValue()
599 TF_DataType dtype = TF_TensorType(t); in Java_org_tensorflow_Tensor_readNDArray()
/external/tensorflow/tensorflow/c/eager/
Dc_api.h163 TF_CAPI_EXPORT extern TF_DataType TFE_TensorHandleDataType(TFE_TensorHandle* h);
316 TF_DataType value);
360 const TF_DataType* values,
/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()
202 TF_DataType_to_PyArray_TYPE(static_cast<TF_DataType>(dtype), &type_num); in ArrayFromMemory()
Dndarray_tensor_bridge.h46 Status TF_DataType_to_PyArray_TYPE(TF_DataType tf_datatype,
Dndarray_tensor.cc31 TF_DataType* out_tf_datatype) { in PyArrayDescr_to_TF_DataType()
70 TF_DataType* out_tf_datatype) { in PyArray_TYPE_to_TF_DataType()
426 TF_DataType dtype = TF_FLOAT; in PyArrayToTF_Tensor()
/external/tensorflow/tensorflow/c/kernels/
Dbitcast_op.cc44 TF_DataType input_data_type;
45 TF_DataType output_data_type;
/external/tensorflow/tensorflow/go/
Dtensor.go35 type DataType C.TF_DataType
90 …c: C.TF_AllocateTensor(C.TF_DataType(dataType), shapePtr, C.int(len(shape)), C.size_t(nbytes)),
129 …c: C.TF_AllocateTensor(C.TF_DataType(dataType), shapePtr, C.int(len(shape)), C.size_t(nbytes)),
216 dataType C.TF_DataType
Dgraph.go414 C.TF_SetAttrType(cdesc, cAttrName, C.TF_DataType(value))
416 var list *C.TF_DataType
418 list = (*C.TF_DataType)(&value[0])
Dattrs.go126 list := make([]C.TF_DataType, meta.list_size)
215 var v C.TF_DataType
/external/tensorflow/tensorflow/python/
Dpywrap_tfe.i104 %typemap(out) TF_DataType {
297 %typemap(out) TF_DataType;

12