Home
last modified time | relevance | path

Searched refs:PadAlignment (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Ddnn.cc134 string PadAlignmentString(PadAlignment alignment) { in PadAlignmentString()
136 case PadAlignment::kDefault: in PadAlignmentString()
138 case PadAlignment::kCudnnPadding: in PadAlignmentString()
140 case PadAlignment::kTensorFlowPadding: in PadAlignmentString()
146 std::ostream& operator<<(std::ostream& str, dnn::PadAlignment alignment) { in operator <<()
Ddnn.h467 enum class PadAlignment : int64 {
474 string PadAlignmentString(PadAlignment alignment);
477 std::ostream& operator<<(std::ostream& str, dnn::PadAlignment alignment);
586 PadAlignment pad_alignment() const { return PadAlignment::kDefault; }
/external/tensorflow/tensorflow/core/util/tensor_bundle/
Dtensor_bundle.cc373 Status PadAlignment(FileOutputBuffer* out, int alignment, int64* size) { in PadAlignment() function
444 status_ = PadAlignment(out_.get(), options_.data_alignment, &size_); in Add()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc657 dnn::PadAlignment::kTensorFlowPadding) in CudnnConvolutionDescriptor()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc658 dnn::PadAlignment::kTensorFlowPadding) { in ScopedConvolutionDescriptor()