/external/tensorflow/tensorflow/core/kernels/ |
D | split_lib_cpu.cc | 31 const Eigen::DSizes<Eigen::DenseIndex, 3>& slice_indices, in operator ()() argument 34 output = input.slice(slice_indices, slice_sizes); in operator ()() 36 output.device(d) = input.slice(slice_indices, slice_sizes); in operator ()() 50 const Eigen::DSizes<Eigen::DenseIndex, 3>& slice_indices, in DEFINE_CPU_KERNELS() 52 output.device(d) = input.slice(slice_indices, slice_sizes); in DEFINE_CPU_KERNELS()
|
D | split_lib.h | 30 const Eigen::DSizes<Eigen::DenseIndex, 2>& slice_indices, 38 const Eigen::DSizes<Eigen::DenseIndex, 3>& slice_indices, 47 const Eigen::DSizes<Eigen::DenseIndex, 3>& slice_indices, 57 const Eigen::DSizes<Eigen::DenseIndex, 3>& slice_indices,
|
D | slice_op.h | 31 const Eigen::DSizes<Eigen::DenseIndex, NDIMS>& slice_indices, in operator() 38 indices[i] = slice_indices[i]; in operator() 46 output.device(d) = input.slice(slice_indices, slice_sizes); in operator()
|
D | split_op.cc | 186 Eigen::DSizes<Eigen::DenseIndex, 3> slice_indices; in Compute() local 189 slice_indices[j] = in Compute() 199 result_shaped = input_reshaped.slice(slice_indices, slice_sizes); in Compute() 204 slice_indices, slice_sizes); in Compute() 337 Eigen::DSizes<Eigen::DenseIndex, 3> slice_indices; in Compute() local 340 slice_indices[j] = indices[j]; in Compute() 349 slice_indices, slice_sizes); in Compute()
|
D | split_lib_gpu.cu.cc | 36 const Eigen::DSizes<Eigen::DenseIndex, 3>& slice_indices, in operator ()() argument 38 To32Bit(output).device(d) = To32Bit(input).slice(slice_indices, slice_sizes); in operator ()() 45 const Eigen::DSizes<Eigen::DenseIndex, 2>& slice_indices, in operator ()() argument 47 To32Bit(output).device(d) = To32Bit(input).slice(slice_indices, slice_sizes); in operator ()()
|
D | batch_util.cc | 159 Eigen::DSizes<Eigen::DenseIndex, NDIMS + 1> slice_indices; in HandleElementToLargerSlice() local 160 slice_indices[0] = index; in HandleElementToLargerSlice() 166 parent_t.slice(slice_indices, slice_size) = element_t.reshape(slice_size); in HandleElementToLargerSlice()
|
D | padding_fifo_queue.cc | 323 Eigen::DSizes<Eigen::DenseIndex, NDIMS + 1> slice_indices; in HandleElementToLargerSlice() local 324 slice_indices[0] = index; in HandleElementToLargerSlice() 330 parent_t.slice(slice_indices, slice_size) = element_t.reshape(slice_size); in HandleElementToLargerSlice()
|
D | topk_op_gpu.cu.cc | 520 const Eigen::DSizes<Eigen::DenseIndex, 2> slice_indices{0, 0}; in LaunchSortKernel() local 523 To32Bit(temp_indices.matrix<int32>()).slice(slice_indices, slice_sizes); in LaunchSortKernel() 525 To32Bit(temp_values.matrix<T>()).slice(slice_indices, slice_sizes); in LaunchSortKernel()
|
D | batch_kernels.cc | 160 Eigen::DSizes<Eigen::DenseIndex, 3> slice_indices{0, position, 0}; in SplitCPU() local 163 output_shaped, input_reshaped, slice_indices, in SplitCPU()
|
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
D | reverse_sequence_op.cc | 121 auto slice_indices = body_builder->DynamicUpdateSlice( in Compile() local 137 auto slice = body_builder->DynamicSlice(output, slice_indices, in Compile()
|
/external/python/cpython2/Objects/ |
D | sliceobject.c | 255 slice_indices(PySliceObject* self, PyObject* len) in slice_indices() function 290 {"indices", (PyCFunction)slice_indices,
|
/external/python/cpython3/Lib/test/ |
D | test_slice.py | 26 def slice_indices(slice, length): function 148 expected = slice_indices(slice, length)
|
D | test_buffer.py | 550 _, _, _, control = slice_indices(s, listlen) 594 _, _, _, slicelen = slice_indices(s_small, small) 1663 self.assertRaises(TypeError, slice_indices, slice(0,1,2), {}) 1664 self.assertRaises(TypeError, slice_indices, "###########", 1) 1665 self.assertRaises(ValueError, slice_indices, slice(0,1,0), 4)
|
/external/python/cpython3/Objects/ |
D | sliceobject.c | 513 slice_indices(PySliceObject* self, PyObject* len) in slice_indices() function 556 {"indices", (PyCFunction)slice_indices,
|
/external/python/cpython3/Modules/ |
D | _testbuffer.c | 1924 slice_indices(PyObject *self, PyObject *args) in slice_indices() function 2807 {"slice_indices", slice_indices, METH_VARARGS, NULL},
|