| /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.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 | ndarray_tensor_bridge.h | 46 Status TF_DataType_to_PyArray_TYPE(TF_DataType tf_datatype,
|
| D | py_func.cc | 384 static_cast<TF_DataType>(t.dtype()), &typenum)); in ConvertTensorToNdarray()
|
| /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/ |
| D | pywrap_tfe.i | 69 %typemap(out) TF_DataType {
197 %typemap(out) 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
|