Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dsplit_lib_cpu.cc31 const Eigen::DSizes<Eigen::DenseIndex, NDims>& 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 ()()
52 const Eigen::DSizes<Eigen::DenseIndex, NDims>& slice_indices, in DEFINE_CPU_KERNELS()
54 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, NDims>& slice_indices,
47 const Eigen::DSizes<Eigen::DenseIndex, NDims>& slice_indices,
57 const Eigen::DSizes<Eigen::DenseIndex, NDims>& 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.cc168 Eigen::DSizes<Eigen::DenseIndex, NDims> slice_indices; in operator ()() local
171 slice_indices[j] = in operator ()()
180 result_shaped = input_reshaped.slice(slice_indices, slice_sizes); in operator ()()
185 input_reshaped, slice_indices, slice_sizes); in operator ()()
378 Eigen::DSizes<Eigen::DenseIndex, 3> slice_indices; in Compute() local
381 slice_indices[j] = indices[j]; in Compute()
390 slice_indices, slice_sizes); in Compute()
Dsplit_lib_gpu.cu.cc37 const Eigen::DSizes<Eigen::DenseIndex, NDims>& slice_indices, in operator ()() argument
39 To32Bit(output).device(d) = To32Bit(input).slice(slice_indices, slice_sizes); in operator ()()
46 const Eigen::DSizes<Eigen::DenseIndex, 2>& slice_indices, in operator ()() argument
48 To32Bit(output).device(d) = To32Bit(input).slice(slice_indices, slice_sizes); in operator ()()
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.h523 const Eigen::DSizes<Eigen::DenseIndex, 2> slice_indices{0, 0};
526 To32Bit(temp_indices.matrix<int32>()).slice(slice_indices, slice_sizes);
528 To32Bit(temp_values.matrix<T>()).slice(slice_indices, slice_sizes);
Dbatch_kernels.cc161 Eigen::DSizes<Eigen::DenseIndex, 2> slice_indices{position, 0}; in SplitCPU() local
165 slice_indices, slice_sizes); in SplitCPU()
/external/tensorflow/tensorflow/core/util/
Dbatch_util.cc216 Eigen::DSizes<Eigen::DenseIndex, NDIMS + 1> slice_indices; in HandleElementToLargerSlice() local
217 slice_indices[0] = index; in HandleElementToLargerSlice()
223 parent_t.slice(slice_indices, slice_size) = element_t.reshape(slice_size); in HandleElementToLargerSlice()
/external/python/cpython3/Lib/test/
Dtest_slice.py26 def slice_indices(slice, length): function
148 expected = slice_indices(slice, length)
Dtest_buffer.py551 _, _, _, control = slice_indices(s, listlen)
595 _, _, _, slicelen = slice_indices(s_small, small)
1664 self.assertRaises(TypeError, slice_indices, slice(0,1,2), {})
1665 self.assertRaises(TypeError, slice_indices, "###########", 1)
1666 self.assertRaises(ValueError, slice_indices, slice(0,1,0), 4)
/external/python/cpython2/Objects/
Dsliceobject.c296 slice_indices(PySliceObject* self, PyObject* len) in slice_indices() function
331 {"indices", (PyCFunction)slice_indices,
/external/python/cpython3/Objects/
Dsliceobject.c514 slice_indices(PySliceObject* self, PyObject* len) in slice_indices() function
557 {"indices", (PyCFunction)slice_indices,
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_instruction_fusion_test.cc554 std::vector<HloInstruction*> slice_indices, update_indices; in TEST_F() local
556 slice_indices.push_back( in TEST_F()
568 slice_indices, in TEST_F()
/external/python/cpython3/Modules/
D_testbuffer.c1922 slice_indices(PyObject *self, PyObject *args) in slice_indices() function
2806 {"slice_indices", slice_indices, METH_VARARGS, NULL},
/external/tensorflow/tensorflow/compiler/xla/service/
Dalgebraic_simplifier_test.cc3843 std::vector<HloInstruction*> slice_indices, update_indices; in TEST_F() local
3845 slice_indices.push_back( in TEST_F()
3857 slice_indices, in TEST_F()