Home
last modified time | relevance | path

Searched refs:set_count (Results 1 – 25 of 28) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/
Dcudnn_pooling_gpu.cc74 input_desc.set_count(in_batch) in Compute()
78 output_desc.set_count(in_batch) in Compute()
193 orig_output_desc.set_count(in_batch) in Compute()
198 orig_input_desc.set_count(in_batch) in Compute()
Dpooling_ops_common.cc239 input_desc.set_count(batch_size) in Compute()
246 output_desc.set_count(batch_size) in Compute()
387 orig_output_desc.set_count(params.tensor_in_batch) in Compute()
394 orig_input_desc.set_count(params.tensor_in_batch) in Compute()
Dfused_batch_norm_op.cc304 x_desc.set_count(batch_size) in operator ()()
311 scale_offset_desc.set_count(1) in operator ()()
458 x_desc.set_count(batch_size) in operator ()()
465 scale_offset_desc.set_count(1) in operator ()()
Dconv_ops_fused_impl.h679 input_desc.set_count(in_batch)
690 bias_desc.set_count(1)
704 output_desc.set_count(out_batch)
Dlrn_op.cc195 dimensions_desc.set_count(batch) in launch()
412 dimensions_desc.set_count(batch) in launch()
Dconv_ops_3d.cc352 input_desc.set_count(in_batch) in launch()
359 output_desc.set_count(in_batch) in launch()
Dconv_grad_ops_3d.cc1255 input_desc.set_count(dims.batch_size) in Compute()
1262 output_desc.set_count(dims.batch_size) in Compute()
1651 input_desc.set_count(dims.batch_size) in Compute()
1661 output_desc.set_count(dims.batch_size) in Compute()
Dconv_ops.cc764 input_desc.set_count(in_batch) in operator ()()
770 output_desc.set_count(out_batch) in operator ()()
Dconv_grad_filter_ops.cc717 input_desc.set_count(dims.batch_size) in operator ()()
723 output_desc.set_count(dims.batch_size) in operator ()()
Dconv_grad_input_ops.cc833 input_desc.set_count(dims.batch_size) in operator ()()
839 output_desc.set_count(dims.batch_size) in operator ()()
/external/grpc-grpc/src/cpp/util/
Dcore_stats.cc34 m->set_count(core.counters[i]); in CoreStatsToProto()
43 b->set_count(core.histograms[grpc_stats_histo_start[i] + j]); in CoreStatsToProto()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_conv_runner.cc185 .set_count(input_shape.dimensions(dnums.input_batch_dimension())); in RunCudnnConvImpl()
222 .set_count(output_shape.dimensions(dnums.output_batch_dimension())); in RunCudnnConvImpl()
274 bias_desc.set_count(1) in RunCudnnConvImpl()
Dcudnn_batchnorm_thunk.cc63 .set_count(batch_size) in MakeDescriptors()
73 .set_count(1); in MakeDescriptors()
/external/perf_data_converter/src/quipper/
Dperf_stat_parser.cc52 newline->set_count(count); in ParsePerfStatOutputToProto()
/external/grpc-grpc/test/cpp/qps/
Dhistogram.h61 p->set_count(grpc_histogram_count(impl_)); in FillProto()
Dclient.h205 rrc->set_count(it->second); in Mark()
Ddriver.cc489 rrc->set_count(it->second); in RunScenario()
/external/perfetto/src/traced/probes/sys_stats/
Dsys_stats_data_source.cc268 irq_stat->set_count(v); in ReadStat()
283 softirq_stat->set_count(v); in ReadStat()
/external/tensorflow/tensorflow/contrib/fused_conv/kernels/
Dfused_conv2d_bias_activation_op.cc582 conv_input_desc.set_count(batch_size) in launch()
594 side_input_desc.set_count(batch_size) in launch()
600 bias_desc.set_count(1) in launch()
606 output_desc.set_count(batch_size) in launch()
/external/v8/src/heap/
Dslot-set.h604 set_count(current_count + 1); in AddSlot()
620 void set_count(int32_t new_value) { in set_count() function
/external/v8/src/objects/
Dcode.h677 inline void set_count(int value);
Dcode-inl.h151 void DependentCode::set_count(int value) { in set_count() function
/external/perfetto/src/profiling/memory/
Dheapprofd_producer.cc556 bucket->set_count(p.second); in Dump()
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h279 BatchDescriptor& set_count(int64 value) {
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc3668 bias_dimensions.set_count(1) in DoBiasAdd()
4103 output_batch_descriptor->set_count(dims[0]) in DeriveOutputBatchDescriptor()

12