Home
last modified time | relevance | path

Searched refs:StreamExecutorConvLayoutsToXlaLayouts (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dstream_executor_util.cc49 StreamExecutorConvLayoutsToXlaLayouts(const ConvolutionDimensionNumbers& dnums, in StreamExecutorConvLayoutsToXlaLayouts() function
129 StreamExecutorConvLayoutsToXlaLayouts(dnums, DataLayout::kBatchDepthYX, in XlaConvLayoutsToStreamExecutorLayouts()
136 StreamExecutorConvLayoutsToXlaLayouts(dnums, DataLayout::kBatchYXDepth, in XlaConvLayoutsToStreamExecutorLayouts()
Dstream_executor_util.h43 StreamExecutorConvLayoutsToXlaLayouts(const ConvolutionDimensionNumbers& dnums,
Dgpu_layout_assignment.cc162 StreamExecutorConvLayoutsToXlaLayouts( in AddBackendConstraintsToDnnConvCustomCall()