Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/
Ddnn.cc165 std::string PadAlignmentString(PadAlignment alignment) { in PadAlignmentString()
167 case PadAlignment::kDefault: in PadAlignmentString()
169 case PadAlignment::kCudnnPadding: in PadAlignmentString()
171 case PadAlignment::kTensorFlowPadding: in PadAlignmentString()
177 std::ostream& operator<<(std::ostream& str, dnn::PadAlignment alignment) { in operator <<()
Ddnn.h459 enum class PadAlignment : int64 { enum
466 std::string PadAlignmentString(PadAlignment alignment);
469 std::ostream& operator<<(std::ostream& str, dnn::PadAlignment alignment);
582 PadAlignment pad_alignment() const { return PadAlignment::kDefault; } in pad_alignment()
/external/tensorflow/tensorflow/core/util/tensor_bundle/
Dtensor_bundle.cc393 Status PadAlignment(FileOutputBuffer* out, int alignment, int64* size) { in PadAlignment() function
471 status_ = PadAlignment(out_.get(), options_.data_alignment, &size_); in Add()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc690 dnn::PadAlignment::kTensorFlowPadding) in CudnnConvolutionDescriptor()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc769 dnn::PadAlignment::kTensorFlowPadding) { in ScopedConvolutionDescriptor()