Searched defs:shard_cost (Results 1 – 9 of 9) 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
251 const int64 shard_cost = in SpatialMaxPool() local
236 const int64 shard_cost = kCostMultiplier * args.out_cols * args.out_depth; in operator ()() local
198 const int64 shard_cost = params.tensor_in_rows * params.tensor_in_cols * in SpatialMaxPoolWithArgMaxHelper() local637 const int64 shard_cost = params.out_width * params.out_height * in SpatialMaxPoolGradGrad() local999 const int64 shard_cost = grad_out->NumElements() / batch_size; in launch() local
347 const int64 shard_cost = in Compute() local
452 const int64 shard_cost = args.in_rows * args.in_cols * args.out_depth; in operator ()() local895 const int64 shard_cost = args.out_rows * args.out_cols * args.out_depth; in operator ()() 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
647 const int64 shard_cost = in launch() local