Searched refs:dst_indices (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | data_format_ops.cc | 66 xla::XlaOp dst_indices = in Compile() local
73 xla::TorchIndexSelect(dst_indices, src_indices, /*dim=*/0); in Compile()
143 std::vector<int32> dst_indices(dim0); in Compile() local
147 dst_indices[j] = i; in Compile()
153 xla::ConstantR1(builder, absl::Span<const int32>(dst_indices)); in Compile()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | pooling_ops_3d.cc | 246 Eigen::DSizes<Eigen::DenseIndex, 5> dst_indices; in launch() local
250 dst.FillIndicesAndSizes<5>(tensor_in.shape(), &dst_indices, in launch()
267 tensor_in.tensor<T, 5>().slice(dst_indices, dst_sizes); in launch()
290 .slice(dst_indices, dst_sizes) in launch()
421 Eigen::DSizes<Eigen::DenseIndex, 5> dst_indices; in launch() local
425 dst.FillIndicesAndSizes<5>(tensor_in_shape, &dst_indices, &dst_sizes); in launch()
444 .slice(dst_indices, dst_sizes) in launch()
|