Searched refs:packed_filters (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | deep_conv2d.cc | 530 std::vector<Tensor>* packed_filters) { in operator ()()
535 auto shard = [&ctx, &packed_filters, &filter_transform_data, &in_depth, in operator ()()
545 &(*packed_filters)[i])); in operator ()()
546 T* packed_filter = (*packed_filters)[i].template flat<T>().data(); in operator ()()
883 const std::vector<Tensor>& packed_filters, const T* input, in operator ()()
900 packed_filters[i].template flat<T>().data(), in operator ()()
979 std::vector<Tensor> packed_filters(tile_spatial_size); in operator ()() local
981 filter_shards_col, filter_transform_data, &packed_filters); in operator ()()
1005 auto shard = [&ctx, &args, &transform, &packed_filters, &in_depth, in operator ()()
1132 num_tiles, packed_filters, input + in_base, in operator ()()
[all …]
|