Searched refs:shard_cost (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | population_count_op.cc | 113 const int64 shard_cost = (total_cost >= static_cast<double>(kint64max)) in operator ()() local
119 shard_cost, shard); in operator ()()
|
| D | compare_and_bitpack_op.cc | 155 const int64 shard_cost = (total_cost >= static_cast<double>(kint64max)) in operator ()() local
161 shard_cost, shard); in operator ()()
|
| D | pooling_ops_common.h | 289 const int64 shard_cost = in SpatialMaxPool() local
292 params.tensor_in_batch, shard_cost, shard); in SpatialMaxPool()
550 const int64 shard_cost =
553 params.tensor_in_batch, shard_cost, shard);
|
| D | depthwise_conv_op.cc | 241 const int64 shard_cost = kCostMultiplier * args.out_cols * args.out_depth; in operator ()() local
245 shard_cost, shard); in operator ()()
|
| D | scatter_functor.h | 174 float shard_cost = kMovingCost * params.dimension(1);
177 Shard(worker_threads.num_threads, worker_threads.workers, N, shard_cost,
|
| D | deep_conv2d.cc | 483 const int64 shard_cost = args.filter_rows * args.filter_cols * in_depth * in operator ()() local
486 Shard(1, worker_threads.workers, out_depth, shard_cost, shard); in operator ()()
1147 const int64 shard_cost = args.out_rows * args.out_cols * args.out_depth * in operator ()() local
1150 shard_cost, shard); in operator ()()
|
| D | maxpooling_op.cc | 207 const int64 shard_cost = params.tensor_in_rows * params.tensor_in_cols * in SpatialMaxPoolWithArgMaxHelper() local
211 params.tensor_in_batch, shard_cost, shard); in SpatialMaxPoolWithArgMaxHelper()
645 const int64 shard_cost = params.out_width * params.out_height * in SpatialMaxPoolGradGrad() local
649 params.tensor_in_batch, shard_cost, shard); in SpatialMaxPoolGradGrad()
1031 const int64 shard_cost = grad_out->NumElements() / batch_size; in launch() local
1033 shard_cost, shard); in launch()
|
| D | avgpooling_op.cc | 384 const int64 shard_cost = in Compute() local
387 out_backprop_batch, shard_cost, shard); in Compute()
|
| D | depthwise_conv_grad_op.cc | 464 const int64 shard_cost = args.in_rows * args.in_cols * args.out_depth; in operator ()() local
467 shard_cost, shard); in operator ()()
936 const int64 shard_cost = args.out_rows * args.out_cols * args.out_depth; in operator ()() local
939 shard_cost, shard); in operator ()()
|
| D | pooling_ops_3d.cc | 643 const int64 shard_cost = in launch() local
647 params.tensor_in_batch, shard_cost, shard); in launch()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_qmatmul_op.cc | 447 float shard_cost = 4 * kArithCost + kMovCost; in GetBiasHandle() local
450 Shard(worker_threads.num_threads, worker_threads.workers, n, shard_cost, in GetBiasHandle()
|