Home
last modified time | relevance | path

Searched refs:output_feature_map_count (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Ddnn.cc431 output_feature_map_count(), input_feature_map_count(), in ToString()
445 std::string od = absl::StrCat("od", output_feature_map_count()); in ToShortString()
471 int64 ret = output_feature_map_count() * input_feature_map_count(); in ComputeWeightCount()
Ddnn.h429 int64 bias_count() const { return output_feature_map_count(); } in bias_count()
431 int64 output_feature_map_count() const { return tensor_.dimensions(0); } in output_feature_map_count() function
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc592 dims[0] = filter_descriptor.output_feature_map_count(); in CudnnFilterDescriptor()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc723 dims[0] = filter_descriptor.output_feature_map_count(); in ScopedFilterDescriptor()