Home
last modified time | relevance | path

Searched refs:TF_EXPORT (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/platform/
Dmacros.h63 #define TF_EXPORT __declspec(dllexport) macro
65 #define TF_EXPORT __declspec(dllimport) macro
68 #define TF_EXPORT __attribute__((visibility("default"))) macro
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dindex_ops_kernel_argmax_float_1d.cc45 extern "C" void TF_EXPORT argmax_float_1d_xla_impl(void* out, void** data) { in argmax_float_1d_xla_impl()
Dindex_ops_kernel_argmax_float_2d.cc50 extern "C" void TF_EXPORT argmax_float_2d_xla_impl(void* out, void** data) { in argmax_float_2d_xla_impl()
/external/tensorflow/tensorflow/core/framework/
Dtypes.h74 TF_EXPORT extern const char* const DEVICE_CPU; // "CPU"
75 TF_EXPORT extern const char* const DEVICE_GPU; // "GPU"
76 TF_EXPORT extern const char* const DEVICE_SYCL; // "SYCL"
Ddataset.h591 TF_EXPORT static const char kDatasetGraphKey[];
595 TF_EXPORT static const char kDatasetGraphOutputNodeKey[];
/external/tensorflow/tensorflow/stream_executor/
Dstream.h1250 TF_EXPORT Stream &ThenBlasGemm(blas::Transpose transa, blas::Transpose transb,
1256 TF_EXPORT Stream &ThenBlasGemm(blas::Transpose transa, blas::Transpose transb,
1261 TF_EXPORT Stream &ThenBlasGemm(blas::Transpose transa, blas::Transpose transb,
1266 TF_EXPORT Stream &ThenBlasGemm(blas::Transpose transa, blas::Transpose transb,
1274 TF_EXPORT Stream &ThenBlasGemm(blas::Transpose transa, blas::Transpose transb,