Home
last modified time | relevance | path

Searched defs:filter_layout (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dstream_executor_util.cc101 std::vector<int64_t> filter_layout; in StreamExecutorConvLayoutsToXlaLayouts() local
223 FilterLayout filter_layout; in XlaConvShapesToStreamExecutorLayouts() local
/external/tensorflow/tensorflow/compiler/xla/stream_executor/
Ddnn.proto102 FilterLayout filter_layout = 4; field
/external/tensorflow/tensorflow/dtensor/mlir/expansions/
Dconv_spmd_expander.cc41 Status VerifyConvLayout(const Layout& input_layout, const Layout& filter_layout, in VerifyConvLayout()
/external/tensorflow/tensorflow/core/kernels/
Dconv_grad_input_ops.cc240 se::dnn::FilterLayout filter_layout; in operator ()() local
Dconv_ops_3d.cc397 se::dnn::FilterLayout filter_layout; in launch() local
Dconv_grad_ops_3d.cc1395 se::dnn::FilterLayout filter_layout; in Compute() local
1791 se::dnn::FilterLayout filter_layout; in Compute() local
Dconv_grad_filter_ops.cc859 se::dnn::FilterLayout filter_layout; in operator ()() local
Dconv_ops.cc1025 se::dnn::FilterLayout filter_layout; in operator ()() local
/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcuda_dnn.cc1089 dnn::FilterLayout filter_layout) { in ToCudnnDataType()
1108 cudnnDataType_t GetCudnnDataType(dnn::FilterLayout filter_layout) { in GetCudnnDataType()