HomeSort by relevance Sort by last modified time
    Searched refs:set_feature_map_count (Results 1 - 15 of 15) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
cudnn_pooling_gpu.cc 74 .set_feature_map_count(in_features)
78 .set_feature_map_count(in_features)
194 .set_feature_map_count(in_features)
199 .set_feature_map_count(in_features)
pooling_ops_common.cc 201 .set_feature_map_count(params.depth)
208 .set_feature_map_count(params.depth)
344 .set_feature_map_count(params.depth)
351 .set_feature_map_count(params.depth)
fused_batch_norm_op.cc 284 .set_feature_map_count(channels)
291 .set_feature_map_count(channels)
438 .set_feature_map_count(channels)
445 .set_feature_map_count(channels)
    [all...]
conv_ops_3d.cc 314 .set_feature_map_count(in_depth)
324 .set_feature_map_count(out_depth)
lrn_op.cc 194 .set_feature_map_count(depth)
411 .set_feature_map_count(depth)
conv_grad_ops_3d.cc 567 .set_feature_map_count(in_depth)
574 .set_feature_map_count(out_depth)
    [all...]
conv_ops.cc 633 .set_feature_map_count(in_depths)
641 .set_feature_map_count(out_depths)
    [all...]
conv_grad_filter_ops.cc     [all...]
conv_grad_input_ops.cc     [all...]
  /external/tensorflow/tensorflow/contrib/fused_conv/kernels/
fused_conv2d_bias_activation_op.cc 446 .set_feature_map_count(conv_input_depth)
460 .set_feature_map_count(output_depth)
466 .set_feature_map_count(output_depth)
472 .set_feature_map_count(output_depth)
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
cudnn_batchnorm_thunk.cc 65 .set_feature_map_count(shape.dimensions(feature_index))
71 .set_feature_map_count(input_desc.feature_map_count())
cudnn_convolution_runner.cc 120 .set_feature_map_count(
155 .set_feature_map_count(
  /external/tensorflow/tensorflow/stream_executor/
dnn.cc 348 output.set_feature_map_count(feature_map_count);
dnn.h 208 // BatchDescriptor().set_count(42).set_feature_map_count(7)...
271 BatchDescriptor& set_feature_map_count(int64 value) { function in class:perftools::gputools::dnn::BatchDescriptor
    [all...]
  /external/tensorflow/tensorflow/stream_executor/cuda/
cuda_dnn.cc     [all...]

Completed in 411 milliseconds