Searched defs:shard_cost (Results 1 – 10 of 10) sorted by relevance
113 const int64 shard_cost = (total_cost >= static_cast<double>(kint64max)) in operator ()() local
155 const int64 shard_cost = (total_cost >= static_cast<double>(kint64max)) in operator ()() local
289 const int64 shard_cost = in SpatialMaxPool() local
241 const int64 shard_cost = kCostMultiplier * args.out_cols * args.out_depth; in operator ()() local
207 const int64 shard_cost = params.tensor_in_rows * params.tensor_in_cols * in SpatialMaxPoolWithArgMaxHelper() local645 const int64 shard_cost = params.out_width * params.out_height * in SpatialMaxPoolGradGrad() local1031 const int64 shard_cost = grad_out->NumElements() / batch_size; in launch() local
464 const int64 shard_cost = args.in_rows * args.in_cols * args.out_depth; in operator ()() local936 const int64 shard_cost = args.out_rows * args.out_cols * args.out_depth; in operator ()() local
384 const int64 shard_cost = in Compute() local
483 const int64 shard_cost = args.filter_rows * args.filter_cols * in_depth * in operator ()() local1147 const int64 shard_cost = args.out_rows * args.out_cols * args.out_depth * in operator ()() local
643 const int64 shard_cost = in launch() local
447 float shard_cost = 4 * kArithCost + kMovCost; in GetBiasHandle() local