Searched refs:FillIndicesAndSizes (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor_slice.h | 121 void FillIndicesAndSizes(
199 void TensorSlice::FillIndicesAndSizes( in FillIndicesAndSizes() function
|
| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_slice_util.h | 179 rel_s.FillIndicesAndSizes<kTensorSliceMaxRank>(shp_s, &s_start, &s_len);
180 rel_d.FillIndicesAndSizes<kTensorSliceMaxRank>(shp_d, &d_start, &d_len);
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | pooling_ops_3d.cc | 252 src.FillIndicesAndSizes<5>(out_backprop.shape(), &src_indices, in launch()
254 dst.FillIndicesAndSizes<5>(tensor_in.shape(), &dst_indices, in launch()
427 src.FillIndicesAndSizes<5>(out_backprop.shape(), &src_indices, in launch()
429 dst.FillIndicesAndSizes<5>(tensor_in_shape, &dst_indices, &dst_sizes); in launch()
|