Searched refs:TF_UINT64 (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | util.cc | 94 return TF_UINT64; in GetTensorFlowDataType()
128 case TF_UINT64: in GetTensorFlowLiteType()
|
| D | util_test.cc | 125 EXPECT_EQ(TF_UINT64, GetTensorFlowDataType(kTfLiteUInt64)); in TEST()
142 EXPECT_EQ(kTfLiteUInt64, GetTensorFlowLiteType(TF_UINT64)); in TEST()
|
| /external/tensorflow/tensorflow/c/ |
| D | tf_datatype.h | 71 TF_UINT64 = 23, enumerator
|
| D | c_api_test.cc | 1515 TF_AllocateTensor(TF_UINT64, dims, 2, 6 * TF_DataTypeSize(TF_UINT64)); in TEST()
1517 TF_AllocateTensor(TF_UINT64, nullptr, 0, TF_DataTypeSize(TF_UINT64)); in TEST()
1523 EXPECT_EQ(6 * TF_DataTypeSize(TF_UINT64), TF_TensorByteSize(a)); in TEST()
1524 EXPECT_EQ(TF_DataTypeSize(TF_UINT64), TF_TensorByteSize(b)); in TEST()
1528 TF_TensorBitcastFrom(a, TF_UINT64, b, new_dims, 2, status); in TEST()
1534 EXPECT_EQ(6 * TF_DataTypeSize(TF_UINT64), TF_TensorByteSize(a)); in TEST()
1535 EXPECT_EQ(6 * TF_DataTypeSize(TF_UINT64), TF_TensorByteSize(b)); in TEST()
|
| D | kernels_test.cc | 898 TEST_KERNEL_TYPE_CONSTRAINT(TF_UINT64, DT_UINT64);
|
| /external/tensorflow/tensorflow/cc/experimental/base/tests/ |
| D | tensor_types_test_util.h | 71 static constexpr TF_DataType kDType = TF_UINT64;
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | dlpack.cc | 96 case TF_DataType::TF_UINT64: in GetDlDataType()
170 *tf_dtype = TF_DataType::TF_UINT64; in TfDataTypeFormDlDataType()
|
| /external/tensorflow/tensorflow/python/lib/core/ |
| D | ndarray_tensor_bridge.cc | 149 case TF_UINT64: in TF_DataType_to_PyArray_TYPE()
|
| D | ndarray_tensor.cc | 146 *out_tf_datatype = TF_UINT64; in PyArray_TYPE_to_TF_DataType()
185 *out_tf_datatype = TF_UINT64; in PyArray_TYPE_to_TF_DataType()
|
| /external/tensorflow/tensorflow/go/ |
| D | tensor.go | 57 Uint64 DataType = C.TF_UINT64
406 {reflect.TypeOf(uint64(0)), C.TF_UINT64},
|
| /external/tensorflow/tensorflow/python/client/ |
| D | tf_session_wrapper.cc | 1235 .value("TF_UINT64", TF_UINT64) in PYBIND11_MODULE()
|