/external/tensorflow/tensorflow/c/ |
D | c_api.h | 98 typedef enum TF_DataType { enum 123 } TF_DataType; typedef 128 TF_CAPI_EXPORT extern size_t TF_DataTypeSize(TF_DataType dt); 234 TF_DataType, const int64_t* dims, int num_dims, void* data, size_t len, 247 TF_CAPI_EXPORT extern TF_Tensor* TF_AllocateTensor(TF_DataType, 259 TF_CAPI_EXPORT extern TF_DataType TF_TensorType(const TF_Tensor*); 513 TF_DataType value); 516 const TF_DataType* values, 592 TF_CAPI_EXPORT extern TF_DataType TF_OperationOutputType(TF_Output oper_out); 598 TF_CAPI_EXPORT extern TF_DataType TF_OperationInputType(TF_Input oper_in); [all …]
|
D | c_api_internal.h | 56 TF_DataType dtype; 185 static Tensor MakeTensor(TF_DataType type, const TensorShape& shape, in MakeTensor() 209 const TF_DataType* types,
|
D | c_api_test.cc | 272 EXPECT_EQ(TF_FLOAT, static_cast<TF_DataType>(tensorflow::DT_FLOAT)); in TEST() 273 EXPECT_EQ(TF_DOUBLE, static_cast<TF_DataType>(tensorflow::DT_DOUBLE)); in TEST() 274 EXPECT_EQ(TF_INT32, static_cast<TF_DataType>(tensorflow::DT_INT32)); in TEST() 275 EXPECT_EQ(TF_UINT8, static_cast<TF_DataType>(tensorflow::DT_UINT8)); in TEST() 276 EXPECT_EQ(TF_INT16, static_cast<TF_DataType>(tensorflow::DT_INT16)); in TEST() 277 EXPECT_EQ(TF_INT8, static_cast<TF_DataType>(tensorflow::DT_INT8)); in TEST() 278 EXPECT_EQ(TF_STRING, static_cast<TF_DataType>(tensorflow::DT_STRING)); in TEST() 279 EXPECT_EQ(TF_COMPLEX64, static_cast<TF_DataType>(tensorflow::DT_COMPLEX64)); in TEST() 281 EXPECT_EQ(TF_INT64, static_cast<TF_DataType>(tensorflow::DT_INT64)); in TEST() 282 EXPECT_EQ(TF_BOOL, static_cast<TF_DataType>(tensorflow::DT_BOOL)); in TEST() [all …]
|
D | c_api.cc | 100 size_t TF_DataTypeSize(TF_DataType dt) { in TF_DataTypeSize() 179 TF_Tensor* TF_AllocateTensor(TF_DataType dtype, const int64_t* dims, in TF_AllocateTensor() 186 TF_Tensor* TF_NewTensor(TF_DataType dtype, const int64_t* dims, int num_dims, in TF_NewTensor() 241 TF_DataType TF_TensorType(const TF_Tensor* t) { return t->dtype; } in TF_TensorType() 517 static TF_Tensor* EmptyTensor(TF_DataType dtype, const TensorShape& shape) { in EmptyTensor() 542 return EmptyTensor(static_cast<TF_DataType>(src.dtype()), src.shape()); in TF_TensorFromTensor() 563 return new TF_Tensor{static_cast<TF_DataType>(src.dtype()), src.shape(), in TF_TensorFromTensor() 681 const TF_DataType* types, in TF_GraphSetOutputHandleShapesAndTypes() 781 EmptyTensor(static_cast<TF_DataType>(src.dtype()), src.shape()); in TF_Run_Helper() 1204 TF_DataType value) { in TF_SetAttrType() [all …]
|
D | c_test_util.h | 85 TF_Operation* RandomUniform(TF_Operation* shape, TF_DataType dtype,
|
D | c_test_util.cc | 229 TF_Operation* RandomUniform(TF_Operation* shape, TF_DataType dtype, in RandomUniform()
|
/external/tensorflow/tensorflow/java/src/main/native/ |
D | tensor_jni.cc | 38 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() 490 static_assert(sizeof(jint) >= sizeof(TF_DataType), in Java_org_tensorflow_Tensor_dtype() 521 TF_DataType dtype = TF_TensorType(t); in Java_org_tensorflow_Tensor_setValue() 592 TF_DataType dtype = TF_TensorType(t); in Java_org_tensorflow_Tensor_readNDArray()
|
D | operation_builder_jni.cc | 201 DEFINE_SET_ATTR(Type, Int, jint, TF_DataType);
|
/external/tensorflow/tensorflow/c/eager/ |
D | c_api.cc | 153 TF_DataType TFE_TensorHandleDataType(TFE_TensorHandle* h) { in TFE_TensorHandleDataType() 154 return static_cast<TF_DataType>(h->t.dtype()); in TFE_TensorHandleDataType() 357 void TFE_OpSetAttrType(TFE_Op* op, const char* attr_name, TF_DataType value) { in TFE_OpSetAttrType() 410 const TF_DataType* values, int num_values) { in TFE_OpSetAttrTypeList() 569 TFE_Op* op, std::vector<TF_DataType>* const_input_types, in OpToFunction() 570 std::vector<TF_DataType>* arg_input_types, in OpToFunction() 619 static_cast<TF_DataType>(op->inputs[i].dtype())); in OpToFunction() 631 static_cast<TF_DataType>(op->inputs[i].dtype())); in OpToFunction() 703 std::vector<TF_DataType> const_input_types; in BuildXlaLaunch() 704 std::vector<TF_DataType> arg_input_types; in BuildXlaLaunch() [all …]
|
D | c_api.h | 121 TF_CAPI_EXPORT extern TF_DataType TFE_TensorHandleDataType(TFE_TensorHandle* h); 195 TF_DataType value); 230 const TF_DataType* values,
|
/external/tensorflow/tensorflow/python/eager/ |
D | pywrap_tensor.cc | 66 TF_DataType src_type_enum, in EagerCast() 67 TF_DataType dst_type_enum, TF_Status* out_status) { in EagerCast() 154 PyObject* PyIntFromDataType(TF_DataType l) { in PyIntFromDataType() 227 static_cast<TF_DataType>(desired_dtype), &desired_np_dtype) in EagerTensor_init() 270 TF_DataType handle_dtype = TFE_TensorHandleDataType(handle.get()); in EagerTensor_init() 275 static_cast<TF_DataType>(desired_dtype), out_status.get())); in EagerTensor_init()
|
D | pywrap_tfe_src.cc | 170 reinterpret_cast<const TF_DataType*>(values.get()), in SetOpAttrList() 305 reinterpret_cast<const TF_DataType*>(values.get()), in SetOpAttrListDefault() 380 TFE_OpSetAttrType(op, key, static_cast<TF_DataType>(value)); in SetOpAttrScalar() 470 static_cast<TF_DataType>(default_value.type())); in SetOpAttrScalarDefault() 1673 tensorflow::gtl::InlinedVector<TF_DataType, 4> attr_value(len);
|
/external/tensorflow/tensorflow/python/lib/core/ |
D | ndarray_tensor_bridge.cc | 109 Status TF_DataType_to_PyArray_TYPE(TF_DataType tf_datatype, in TF_DataType_to_PyArray_TYPE() 201 TF_DataType_to_PyArray_TYPE(static_cast<TF_DataType>(dtype), &type_num); in ArrayFromMemory()
|
D | ndarray_tensor_bridge.h | 46 Status TF_DataType_to_PyArray_TYPE(TF_DataType tf_datatype,
|
D | ndarray_tensor.cc | 31 TF_DataType* out_tf_datatype) { in PyArrayDescr_to_TF_DataType() 70 TF_DataType* out_tf_datatype) { in PyArray_TYPE_to_TF_DataType() 383 TF_DataType dtype = TF_FLOAT; in PyArrayToTF_Tensor()
|
D | py_func.cc | 384 static_cast<TF_DataType>(t.dtype()), &typenum)); in ConvertTensorToNdarray()
|
/external/tensorflow/tensorflow/python/ |
D | pywrap_tfe.i | 69 %typemap(out) TF_DataType { 197 %typemap(out) TF_DataType;
|
/external/tensorflow/tensorflow/go/ |
D | graph.go | 305 C.TF_SetAttrType(cdesc, cAttrName, C.TF_DataType(value)) 307 var list *C.TF_DataType 309 list = (*C.TF_DataType)(&value[0])
|
D | tensor.go | 35 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)), 220 dataType C.TF_DataType
|
/external/tensorflow/tensorflow/python/client/ |
D | tf_session.i | 697 %typemap(in) (const std::vector<TF_DataType>& types) 698 (std::vector<TF_DataType> types_local){ 715 types_local.push_back((TF_DataType) PyInt_AsLong(item));
|
D | tf_session_helper.h | 196 const std::vector<int>& ranks, const std::vector<TF_DataType>& types,
|
D | tf_session_helper.cc | 452 const std::vector<int>& ranks, const std::vector<TF_DataType>& types, in TF_GraphSetOutputHandleShapesAndTypes_wrapper()
|
/external/tensorflow/tensorflow/docs_src/extend/ |
D | language_bindings.md | 96 TensorFlow. Corresponds to `TF_DataType` in the C API and often referred to
|