Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32_grad/
Darithmetic_grad.c56 int input_iter[8] = {0}; in MaximumByAxes() local
60 size_t offset = GetInputOffset(num_dims, input0_dims, input_iter); in MaximumByAxes()
63 } while (NextIndex(num_dims, input0_dims, input_iter)); in MaximumByAxes()
68 int input_iter[8] = {0}; in MaximumByAxes() local
83 size_t offset0 = GetOutputOffset(num_dims, input0_dims, input_iter, num_axes0, axes0); in MaximumByAxes()
84 size_t offset1 = GetOutputOffset(num_dims, input1_dims, input_iter, num_axes1, axes1); in MaximumByAxes()
85 size_t yt_offset = GetInputOffset(num_dims, input0_dims, input_iter); in MaximumByAxes()
88 } while (NextIndex(num_dims, dy_dims, input_iter)); in MaximumByAxes()
106 int input_iter[8] = {0}; in MinimumByAxes() local
110 size_t offset = GetInputOffset(num_dims, input0_dims, input_iter); in MinimumByAxes()
[all …]
Dreduce_grad.c20 void ReduceMeanByAxes(const float *input_data, int *input_iter, const int *input_dims, int input_nu… in ReduceMeanByAxes() argument
30 input_iter[idx] = 0; in ReduceMeanByAxes()
34 size_t input_offset = GetInputOffset(input_num_dims, input_dims, input_iter); in ReduceMeanByAxes()
35 size_t output_offset = GetOutputOffset(input_num_dims, input_dims, input_iter, num_axes, axes); in ReduceMeanByAxes()
37 } while (NextIndex(input_num_dims, input_dims, input_iter)); in ReduceMeanByAxes()
73 int input_iter[8] = {0}; in ReduceSumByAxes() local
84 size_t input_offset = GetInputOffset(num_dims, input_dims, input_iter); in ReduceSumByAxes()
85 size_t output_offset = GetOutputOffset(num_dims, input_dims, input_iter, num_axes, axes); in ReduceSumByAxes()
87 } while (NextIndex(num_dims, input_dims, input_iter)); in ReduceSumByAxes()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16_grad/
Darithmetic_grad.c58 int input_iter[8] = {0}; in MaximumByAxesFp16() local
62 size_t offset = GetInputOffset(num_dims, input0_dims, input_iter); in MaximumByAxesFp16()
65 } while (NextIndex(num_dims, input0_dims, input_iter)); in MaximumByAxesFp16()
70 int input_iter[8] = {0}; in MaximumByAxesFp16() local
85 size_t offset0 = GetOutputOffset(num_dims, input0_dims, input_iter, num_axes0, axes0); in MaximumByAxesFp16()
86 size_t offset1 = GetOutputOffset(num_dims, input1_dims, input_iter, num_axes1, axes1); in MaximumByAxesFp16()
87 size_t yt_offset = GetInputOffset(num_dims, input0_dims, input_iter); in MaximumByAxesFp16()
90 } while (NextIndex(num_dims, dy_dims, input_iter)); in MaximumByAxesFp16()
109 int input_iter[8] = {0}; in MinimumByAxesFp16() local
113 size_t offset = GetInputOffset(num_dims, input0_dims, input_iter); in MinimumByAxesFp16()
[all …]
/third_party/boost/libs/stl_interfaces/test/
Dinput.cpp55 struct input_iter : boost::stl_interfaces::iterator_interface< struct
56 input_iter<ValueType>,
60 input_iter() : it_(nullptr) {} in input_iter() function
61 input_iter(ValueType * it) : it_(it) {} in input_iter() argument
66 input_iter(input_iter<ValueType2> it) : it_(it.it_) in input_iter() argument
70 input_iter & operator++() noexcept in operator ++() argument
75 friend bool operator==(input_iter lhs, input_iter rhs) noexcept in operator ==()
81 input_iter<ValueType>, argument
90 friend struct input_iter; argument
93 BOOST_STL_INTERFACES_STATIC_ASSERT_CONCEPT(input_iter<int>, std::input_iterator)
[all …]
/third_party/boost/libs/iterator/test/
Diterator_facade.cpp69 struct input_iter struct
71 input_iter
78 input_iter() {} in input_iter() function
89 bool equal(input_iter const&) const in equal() argument
201 input_iter p; in main()
205 same_type<input_iter::pointer>(p.operator->()); in main()
Diterator_adaptor_test.cpp280 boost::input_iterator_archetype_no_proxy<dummyT> input_iter; in main() local
282 adaptor_type i(input_iter); in main()
/third_party/boost/boost/compute/iterator/detail/
Dswizzle_iterator.hpp70 swizzle_iterator_index_expr(const InputIterator &input_iter, in swizzle_iterator_index_expr()
73 : m_input_iter(input_iter), in swizzle_iterator_index_expr()
/third_party/boost/libs/compute/include/boost/compute/iterator/detail/
Dswizzle_iterator.hpp70 swizzle_iterator_index_expr(const InputIterator &input_iter, in swizzle_iterator_index_expr()
73 : m_input_iter(input_iter), in swizzle_iterator_index_expr()
/third_party/boost/libs/compute/include/boost/compute/iterator/
Dtransform_iterator.hpp70 transform_iterator_index_expr(const InputIterator &input_iter, in transform_iterator_index_expr()
73 : m_input_iter(input_iter), in transform_iterator_index_expr()
Dstrided_iterator.hpp86 strided_iterator_index_expr(const Iterator &input_iter, in strided_iterator_index_expr()
89 : m_input_iter(input_iter), in strided_iterator_index_expr()
/third_party/boost/boost/compute/iterator/
Dtransform_iterator.hpp70 transform_iterator_index_expr(const InputIterator &input_iter, in transform_iterator_index_expr()
73 : m_input_iter(input_iter), in transform_iterator_index_expr()
Dstrided_iterator.hpp86 strided_iterator_index_expr(const Iterator &input_iter, in strided_iterator_index_expr()
89 : m_input_iter(input_iter), in strided_iterator_index_expr()
/third_party/python/Modules/
D_csv.c107 PyObject *input_iter; /* iterate over this for input lines */ member
878 lineobj = PyIter_Next(self->input_iter); in Reader_iternext()
953 Py_VISIT(self->input_iter); in Reader_traverse()
963 Py_CLEAR(self->input_iter); in Reader_clear()
1022 self->input_iter = NULL; in csv_reader()
1036 self->input_iter = PyObject_GetIter(iterator); in csv_reader()
1037 if (self->input_iter == NULL) { in csv_reader()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/mem_reuse/
Dmem_reuse.cc339 auto input_iter = kernel_map_.find(input_key); in SetKernelDefInputs() local
340 if (input_iter == kernel_map_.end()) { in SetKernelDefInputs()
343 kernel_def->InsertInputKernel(input_iter->second); in SetKernelDefInputs()
/third_party/mindspore/mindspore/ccsrc/vm/
Dgraph_partition.cc435 auto input_iter = node_to_segment.find(input); in AddSegmentDependency() local
436 if (input_iter != node_to_segment.end()) { in AddSegmentDependency()
437 input_segment = input_iter->second; in AddSegmentDependency()