Home
last modified time | relevance | path

Searched refs:filter_layout (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dstream_executor_util.cc101 std::vector<int64_t> filter_layout; in StreamExecutorConvLayoutsToXlaLayouts() local
104 filter_layout.push_back(dnums.kernel_output_feature_dimension()); in StreamExecutorConvLayoutsToXlaLayouts()
105 filter_layout.push_back(dnums.kernel_input_feature_dimension()); in StreamExecutorConvLayoutsToXlaLayouts()
106 filter_layout.insert(filter_layout.end(), in StreamExecutorConvLayoutsToXlaLayouts()
112 filter_layout.push_back(dnums.kernel_output_feature_dimension()); in StreamExecutorConvLayoutsToXlaLayouts()
113 filter_layout.push_back(dnums.kernel_input_feature_dimension()); in StreamExecutorConvLayoutsToXlaLayouts()
114 filter_layout.insert(filter_layout.end(), in StreamExecutorConvLayoutsToXlaLayouts()
117 filter_layout.push_back(FindMissingDnum(filter_layout)); in StreamExecutorConvLayoutsToXlaLayouts()
120 filter_layout.push_back(dnums.kernel_output_feature_dimension()); in StreamExecutorConvLayoutsToXlaLayouts()
121 filter_layout.insert(filter_layout.end(), in StreamExecutorConvLayoutsToXlaLayouts()
[all …]
/external/tensorflow/tensorflow/dtensor/mlir/expansions/
Dconv_spmd_expander.cc41 Status VerifyConvLayout(const Layout& input_layout, const Layout& filter_layout, in VerifyConvLayout() argument
43 if (!filter_layout.IsFullyReplicated()) in VerifyConvLayout()
143 TF_ASSIGN_OR_RETURN(const Layout filter_layout, in HandleConv()
148 TF_RETURN_IF_ERROR(VerifyConvLayout(input_layout, filter_layout, conv_op)); in HandleConv()
/external/tensorflow/tensorflow/core/kernels/
Dconv_grad_input_ops.cc240 se::dnn::FilterLayout filter_layout; in operator ()() local
242 std::tie(compute_data_layout, filter_layout) = in operator ()()
262 .set_layout(filter_layout); in operator ()()
Dconv_ops_3d.cc397 se::dnn::FilterLayout filter_layout; in launch() local
399 std::tie(compute_data_layout, filter_layout) = in launch()
425 .set_layout(filter_layout); in launch()
Dconv_ops_fused_impl.h488 se::dnn::FilterLayout filter_layout;
489 std::tie(compute_data_layout, filter_layout) =
503 .set_layout(filter_layout);
Dconv_grad_ops_3d.cc1395 se::dnn::FilterLayout filter_layout; in Compute() local
1397 std::tie(compute_data_layout, filter_layout) = in Compute()
1420 .set_layout(filter_layout); in Compute()
1791 se::dnn::FilterLayout filter_layout; in Compute() local
1793 std::tie(compute_data_layout, filter_layout) = in Compute()
1819 .set_layout(filter_layout); in Compute()
Dconv_grad_filter_ops.cc859 se::dnn::FilterLayout filter_layout; in operator ()() local
861 std::tie(compute_data_layout, filter_layout) = in operator ()()
881 .set_layout(filter_layout); in operator ()()
Dconv_ops.cc1025 se::dnn::FilterLayout filter_layout; in operator ()() local
1027 std::tie(compute_data_layout, filter_layout) = in operator ()()
1047 .set_layout(filter_layout); in operator ()()
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/include/mlir-hlo/Dialect/lhlo_gpu/IR/
Dlhlo_gpu_ops_enums.td75 // Note: Not using names like input_layout or filter_layout as `input` may be
/external/tensorflow/tensorflow/compiler/xla/stream_executor/
Ddnn.proto102 FilterLayout filter_layout = 4; field
Ddnn.h463 FilterLayout layout() const { return tensor_.filter_layout(); } in layout()
/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcuda_dnn.cc1089 dnn::FilterLayout filter_layout) { in ToCudnnDataType() argument
1091 filter_layout == dnn::FilterLayout::kOutputInputYX4) { in ToCudnnDataType()
1095 filter_layout == dnn::FilterLayout::kOutputInputYX32) { in ToCudnnDataType()
1108 cudnnDataType_t GetCudnnDataType(dnn::FilterLayout filter_layout) { in GetCudnnDataType() argument
1109 return ToCudnnDataType(dnn::ToDataType<T>::value, filter_layout); in GetCudnnDataType()