| /external/tensorflow/tensorflow/core/kernels/ |
| split_lib_cpu.cc | 32 const Eigen::DSizes<Eigen::DenseIndex, 3>& slice_sizes) {
34 output = input.slice(slice_indices, slice_sizes);
36 output.device(d) = input.slice(slice_indices, slice_sizes);
51 const Eigen::DSizes<Eigen::DenseIndex, 3>& slice_sizes) {
52 output.device(d) = input.slice(slice_indices, slice_sizes);
|
| split_lib.h | 31 const Eigen::DSizes<Eigen::DenseIndex, 2>& slice_sizes);
39 const Eigen::DSizes<Eigen::DenseIndex, 3>& slice_sizes);
48 const Eigen::DSizes<Eigen::DenseIndex, 3>& slice_sizes);
58 const Eigen::DSizes<Eigen::DenseIndex, 3>& slice_sizes);
|
| slice_op.h | 32 const Eigen::DSizes<Eigen::DenseIndex, NDIMS>& slice_sizes) {
42 sizes[i] = slice_sizes[i];
46 output.device(d) = input.slice(slice_indices, slice_sizes);
|
| split_op.cc | 187 Eigen::DSizes<Eigen::DenseIndex, 3> slice_sizes;
191 slice_sizes[j] = sizes[j];
199 result_shaped = input_reshaped.slice(slice_indices, slice_sizes);
204 slice_indices, slice_sizes); local
338 Eigen::DSizes<Eigen::DenseIndex, 3> slice_sizes;
341 slice_sizes[j] = sizes[j];
349 slice_indices, slice_sizes);
|
| split_lib_gpu.cu.cc | 37 const Eigen::DSizes<Eigen::DenseIndex, 3>& slice_sizes) {
38 To32Bit(output).device(d) = To32Bit(input).slice(slice_indices, slice_sizes);
46 const Eigen::DSizes<Eigen::DenseIndex, 2>& slice_sizes) {
47 To32Bit(output).device(d) = To32Bit(input).slice(slice_indices, slice_sizes);
|
| topk_op_gpu.cu.cc | 521 const Eigen::DSizes<Eigen::DenseIndex, 2> slice_sizes{num_rows, k};
523 To32Bit(temp_indices.matrix<int32>()).slice(slice_indices, slice_sizes);
525 To32Bit(temp_values.matrix<T>()).slice(slice_indices, slice_sizes);
|
| batch_kernels.cc | 161 Eigen::DSizes<Eigen::DenseIndex, 3> slice_sizes{1, size, suffix_dim_size};
164 slice_sizes);
365 const std::vector<int64> slice_sizes = {1}; local
373 slice_sizes, &slices); \
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| dynamic_ops_test.cc | 124 const std::vector<int64>& slice_sizes,
147 builder.DynamicSlice(input, starts, slice_sizes);
155 const std::vector<int64>& slice_sizes,
173 builder.DynamicSlice(input, starts, slice_sizes);
181 const std::vector<int64>& slice_sizes,
199 builder.DynamicSlice(input, starts, slice_sizes);
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| shape_inference.h | 167 // in 'slice_sizes', with dynamic start indices shape 'start_indices_shape'.
170 tensorflow::gtl::ArraySlice<int64> slice_sizes);
|
| shape_inference.cc | [all...] |
| hlo_instruction.h | 343 // 'slice_sizes'.
347 tensorflow::gtl::ArraySlice<int64> slice_sizes);
1030 int64 slice_sizes(int64 dimension) const { function in class:xla::HloInstruction
[all...] |
| hlo_instruction.cc | 520 tensorflow::gtl::ArraySlice<int64> slice_sizes) {
525 instruction->dynamic_slice_sizes_.assign(slice_sizes.begin(),
526 slice_sizes.end());
[all...] |
| user_computation.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| local_computation_builder.cc | 374 tensorflow::gtl::ArraySlice<int64> slice_sizes) {
375 return builder_.DynamicSlice(operand, start_indices, slice_sizes);
|
| local_computation_builder.h | 176 tensorflow::gtl::ArraySlice<int64> slice_sizes);
|
| xla_client.py | 780 def DynamicSlice(self, operand, start_indices, slice_sizes):
787 slice_sizes: iterable of N integers containing the slice sizes in each
797 slice_sizes))
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/ |
| xla_data.proto | 449 repeated int64 slice_sizes = 4;
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| computation_builder.cc | 318 tensorflow::gtl::ArraySlice<int64> slice_sizes) {
323 for (int64 index : slice_sizes) {
[all...] |
| computation_builder.h | 256 // The size of the slice in each dimension is passed in 'slice_sizes',
266 tensorflow::gtl::ArraySlice<int64> slice_sizes);
[all...] |
