Searched refs:shard_cost (Results 1 – 9 of 9) 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 | 251 const int64 shard_cost = in SpatialMaxPool() local
254 params.tensor_in_batch, shard_cost, shard); in SpatialMaxPool()
498 const int64 shard_cost =
501 params.tensor_in_batch, shard_cost, shard);
|
| D | depthwise_conv_op.cc | 236 const int64 shard_cost = kCostMultiplier * args.out_cols * args.out_depth; in operator ()() local
240 shard_cost, shard); in operator ()()
|
| D | maxpooling_op.cc | 198 const int64 shard_cost = params.tensor_in_rows * params.tensor_in_cols * in SpatialMaxPoolWithArgMaxHelper() local
202 params.tensor_in_batch, shard_cost, shard); in SpatialMaxPoolWithArgMaxHelper()
637 const int64 shard_cost = params.out_width * params.out_height * in SpatialMaxPoolGradGrad() local
641 params.tensor_in_batch, shard_cost, shard); in SpatialMaxPoolGradGrad()
999 const int64 shard_cost = grad_out->NumElements() / batch_size; in launch() local
1001 shard_cost, shard); in launch()
|
| 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 | depthwise_conv_grad_op.cc | 452 const int64 shard_cost = args.in_rows * args.in_cols * args.out_depth; in operator ()() local
455 shard_cost, shard); in operator ()()
895 const int64 shard_cost = args.out_rows * args.out_cols * args.out_depth; in operator ()() local
898 shard_cost, shard); in operator ()()
|
| D | avgpooling_op.cc | 347 const int64 shard_cost = in Compute() local
350 out_backprop_batch, shard_cost, shard); in Compute()
|
| D | pooling_ops_3d.cc | 647 const int64 shard_cost = in launch() local
651 params.tensor_in_batch, shard_cost, shard); in launch()
|