Searched refs:grad_offset (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/ps/ |
| D | optimizer_info.cc | 87 size_t grad_offset = 0; in Accumulate() local
89 grad_offset += IntToSize(lengths[i]); in Accumulate()
91 float *grad_data = const_cast<float *>(values.data()) + grad_offset; in Accumulate()
127 size_t grad_offset = 0; in Accumulate() local
129 grad_offset += IntToSize(lengths[i]); in Accumulate()
131 float *incr_grad_data = const_cast<float *>(values.data()) + grad_offset; in Accumulate()
|
| D | worker.cc | 600 int64_t grad_offset = 0; in BuildSparseValue() local
602 grad_offset += lengths[i]; in BuildSparseValue()
607 dst_data = reduced_data->data() + grad_offset; in BuildSparseValue()
617 int64_t indice_offset = grad_offset + lengths[grad_index]; in BuildSparseValue()
745 size_t grad_offset = 0; in SparsePartitioner() local
748 grad_offset += send.len()[i]; in SparsePartitioner()
754 float *grad_data = data + grad_offset; in SparsePartitioner()
|
| /third_party/vk-gl-cts/external/openglcts/modules/glesext/texture_cube_map_array/ |
| D | esextcTextureCubeMapArraySampling.cpp | 1042 const glw::GLuint grad_offset = i * n_grad_components; in prepareGradXForFace() local
1044 …tNeighbour(texture_coordinates + texture_coordinates_offset, face, 1, width, grad_x + grad_offset); in prepareGradXForFace()
1050 const glw::GLuint grad_offset = i * n_grad_components; in prepareGradXForFace() local
1053 grad_x + grad_offset); in prepareGradXForFace()
1059 const glw::GLuint grad_offset = i * n_grad_components; in prepareGradXForFace() local
1061 vectorSubtractInPlace<3>(grad_x + grad_offset, texture_coordinates + texture_coordinates_offset); in prepareGradXForFace()
1081 const glw::GLuint grad_offset = i * n_grad_components; in prepareGradYForFace() local
1083 …pNeighbour(texture_coordinates + texture_coordinates_offset, face, 1, width, grad_y + grad_offset); in prepareGradYForFace()
1089 const glw::GLuint grad_offset = i * n_grad_components; in prepareGradYForFace() local
1091 …ur(texture_coordinates + texture_coordinates_offset, face, 4 * width, width, grad_y + grad_offset); in prepareGradYForFace()
[all …]
|