Searched refs:dst_indices (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | permute_op.cc | 67 std::vector<int32> dst_indices(4, 0); in Compile() local
71 dst_indices[i] = j; in Compile()
76 auto keys = xla::ConstantR1(builder, absl::Span<const int32>(dst_indices)); in Compile()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | pooling_ops_3d.cc | 250 Eigen::DSizes<Eigen::DenseIndex, 5> dst_indices; in launch() local
254 dst.FillIndicesAndSizes<5>(tensor_in.shape(), &dst_indices, in launch()
271 tensor_in.tensor<T, 5>().slice(dst_indices, dst_sizes); in launch()
294 .slice(dst_indices, dst_sizes) in launch()
425 Eigen::DSizes<Eigen::DenseIndex, 5> dst_indices; in launch() local
429 dst.FillIndicesAndSizes<5>(tensor_in_shape, &dst_indices, &dst_sizes); in launch()
448 .slice(dst_indices, dst_sizes) in launch()
|