Home
last modified time | relevance | path

Searched refs:slice_indices (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dsplit_lib_cpu.cc31 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()
Dsplit_lib.h30 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,
Dslice_op.h31 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()
Dsplit_op.cc186 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()
Dsplit_lib_gpu.cu.cc36 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 ()()
Dbatch_util.cc159 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()
Dpadding_fifo_queue.cc323 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()
Dtopk_op_gpu.cu.cc520 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()
Dbatch_kernels.cc160 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/
Dreverse_sequence_op.cc121 auto slice_indices = body_builder->DynamicUpdateSlice( in Compile() local
137 auto slice = body_builder->DynamicSlice(output, slice_indices, in Compile()
/external/python/cpython2/Objects/
Dsliceobject.c255 slice_indices(PySliceObject* self, PyObject* len) in slice_indices() function
290 {"indices", (PyCFunction)slice_indices,
/external/python/cpython3/Lib/test/
Dtest_slice.py26 def slice_indices(slice, length): function
148 expected = slice_indices(slice, length)
Dtest_buffer.py550 _, _, _, 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/
Dsliceobject.c513 slice_indices(PySliceObject* self, PyObject* len) in slice_indices() function
556 {"indices", (PyCFunction)slice_indices,
/external/python/cpython3/Modules/
D_testbuffer.c1924 slice_indices(PyObject *self, PyObject *args) in slice_indices() function
2807 {"slice_indices", slice_indices, METH_VARARGS, NULL},