Home
last modified time | relevance | path

Searched refs:grad_x (Results 1 – 20 of 20) sorted by relevance

/third_party/mindspore/mindspore/core/ops/grad/
Dmaximum_grad.cc22 void MaximumGrad::Init(const bool grad_x, const bool grad_y) { in Init() argument
23 set_grad_x(grad_x); in Init()
27 void MaximumGrad::set_grad_x(const bool grad_x) { (void)this->AddAttr(kGradX, MakeValue(grad_x)); } in set_grad_x() argument
Dminimum_grad.cc22 void MinimumGrad::Init(const bool grad_x, const bool grad_y) { in Init() argument
23 set_grad_x(grad_x); in Init()
27 void MinimumGrad::set_grad_x(const bool grad_x) { (void)this->AddAttr(kGradX, MakeValue(grad_x)); } in set_grad_x() argument
Dminimum_grad.h31 void Init(const bool grad_x = true, const bool grad_y = true);
32 void set_grad_x(const bool grad_x);
Dmaximum_grad.h31 void Init(const bool grad_x = true, const bool grad_y = true);
32 void set_grad_x(const bool grad_x);
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dmirror_pad_grad_cpu_kernel.cc178 const int64_t grad_x = (SizeToLong(pos) % dx_width) + paddings[WIDTH]; in MirrorPadGrad_Width_Height() local
182 dx[pos] = interim_dy[(dx_block_num * dy_height + grad_y) * dy_width + grad_x]; in MirrorPadGrad_Width_Height()
184 int64_t x_dist_1 = (ap1_x - grad_x - mode); in MirrorPadGrad_Width_Height()
186 int64_t x_dist_2 = (ap2_x - grad_x + mode); in MirrorPadGrad_Width_Height()
201 temp_x = grad_x; in MirrorPadGrad_Width_Height()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/
Dmirror_pad_impl.cu189 const int grad_x = (pos % dx_width) + paddings[WIDTH + LEFT]; in MirrorPadGrad_Width_Height() local
192 dx[pos] = interim_dy[(dx_block_num * dy_height + grad_y) * dy_width + grad_x]; in MirrorPadGrad_Width_Height()
194 int x_dist_1 = (ap1_x - grad_x - mode); in MirrorPadGrad_Width_Height()
196 int x_dist_2 = (ap2_x - grad_x + mode); in MirrorPadGrad_Width_Height()
211 temp_x = grad_x; in MirrorPadGrad_Width_Height()
/third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/
Dnn_norm_ops_declare.cc31 OUTPUT_MAP(SoftmaxGrad) = {{0, OUTPUT_DESC(grad_x)}};
Delewise_calculation_ops_declare.cc486 ATTR_MAP(MaximumGrad) = {{"grad_x", ATTR_DESC(grad_x, AnyTraits<bool>())},
493 ATTR_MAP(MinimumGrad) = {{"grad_x", ATTR_DESC(grad_x, AnyTraits<bool>())},
/third_party/mindspore/mindspore/lite/schema/
Dops.fbs694 grad_x: bool;
730 grad_x: bool;
Dops_generated.h7163 bool grad_x() const {
7181 void add_grad_x(bool grad_x) {
7182 fbb_.AddElement<uint8_t>(MaximumGrad::VT_GRAD_X, static_cast<uint8_t>(grad_x), 0);
7200 bool grad_x = false,
7204 builder_.add_grad_x(grad_x);
7571 bool grad_x() const {
7589 void add_grad_x(bool grad_x) {
7590 fbb_.AddElement<uint8_t>(MinimumGrad::VT_GRAD_X, static_cast<uint8_t>(grad_x), 0);
7608 bool grad_x = false,
7612 builder_.add_grad_x(grad_x);
/third_party/skia/third_party/externals/spirv-cross/
Dspirv_hlsl.cpp2846 uint32_t grad_x = 0; in emit_texture_op() local
2873 test(grad_x, ImageOperandsGradMask); in emit_texture_op()
2919 else if (lod || grad_x || grad_y) in emit_texture_op()
2960 else if (grad_x || grad_y) in emit_texture_op()
2996 if (grad_x || grad_y) in emit_texture_op()
3065 if (grad_x || grad_y) in emit_texture_op()
3124 if (!dref && (grad_x || grad_y)) in emit_texture_op()
3126 forward = forward && should_forward(grad_x); in emit_texture_op()
3129 expr += to_expression(grad_x); in emit_texture_op()
Dspirv_glsl.hpp422 uint32_t grad_x = 0, grad_y = 0, lod = 0, coffset = 0, offset = 0; member
Dspirv_msl.cpp9443 uint32_t grad_x = args.grad_x; in to_function_args() local
9731 if (msl_options.is_macos() && (grad_x || grad_y)) in to_function_args()
9737 bool constant_zero_x = !grad_x || expression_is_constant_null(grad_x); in to_function_args()
9742 grad_x = 0; in to_function_args()
9799 if ((grad_x || grad_y) && (imgtype.image.dim != Dim1D || msl_options.texture_1D_as_2D)) in to_function_args()
9801 forward = forward && should_forward(grad_x); in to_function_args()
9823 …farg_str += ", gradient" + grad_opt + "(" + to_expression(grad_x) + ", " + to_expression(grad_y) +… in to_function_args()
Dspirv_glsl.cpp6734 uint32_t grad_x = 0; in to_texture_op() local
6760 test(grad_x, ImageOperandsGradMask); in to_texture_op()
6781 name_args.has_grad = grad_x != 0 || grad_y != 0; in to_texture_op()
6798 args.grad_x = grad_x; in to_texture_op()
7109 if (args.grad_x || args.grad_y) in to_function_args()
7111 forward = forward && should_forward(args.grad_x); in to_function_args()
7114 farg_str += to_expression(args.grad_x); in to_function_args()
/third_party/mindspore/mindspore/lite/src/ops/
Dops_def.cc694 OP_ATTR(grad_x, bool)
730 OP_ATTR(grad_x, bool)
/third_party/vk-gl-cts/external/openglcts/modules/glesext/texture_cube_map_array/
DesextcTextureCubeMapArraySampling.cpp1033 void prepareGradXForFace(glw::GLfloat* grad_x, glw::GLuint face, glw::GLfloat* texture_coordinates,… in prepareGradXForFace() argument
1044 …getRightNeighbour(texture_coordinates + texture_coordinates_offset, face, 1, width, grad_x + grad_… in prepareGradXForFace()
1053 grad_x + grad_offset); in prepareGradXForFace()
1061 vectorSubtractInPlace<3>(grad_x + grad_offset, texture_coordinates + texture_coordinates_offset); in prepareGradXForFace()
3546 buffer = &buffers.grad_x; in setupSharedStorageBuffer()
4096 …grad_x.init(gl, (glw::GLsizeiptr)(grad_x_buffer_data.size() * sizeof(glw::GLfloat)), &grad_x_buffe…
5114 buffer = &buffers.grad_x;
DesextcTextureCubeMapArraySampling.hpp346 bufferDefinition grad_x; member in glcts::TextureCubeMapArraySamplingTest::bufferCollection
/third_party/mindspore/mindspore/ops/operations/
D_grad_ops.py1039 def __init__(self, grad_x=True, grad_y=True): argument
1115 def __init__(self, grad_x=True, grad_y=True): argument
/third_party/mindspore/mindspore/lite/schema/inner/
Dops_generated.h11364 bool grad_x = false;
11378 bool grad_x() const {
11405 void add_grad_x(bool grad_x) {
11406 fbb_.AddElement<uint8_t>(MaximumGrad::VT_GRAD_X, static_cast<uint8_t>(grad_x), 0);
11424 bool grad_x = false,
11428 builder_.add_grad_x(grad_x);
11922 bool grad_x = false;
11936 bool grad_x() const {
11963 void add_grad_x(bool grad_x) {
11964 fbb_.AddElement<uint8_t>(MinimumGrad::VT_GRAD_X, static_cast<uint8_t>(grad_x), 0);
[all …]
/third_party/mindspore/config/
Dop_info.config317 …": false, "param_type": "required", "shape": "all"}], "attr": [{"name": "grad_x", "param_type": "o…
318 …": false, "param_type": "required", "shape": "all"}], "attr": [{"name": "grad_x", "param_type": "o…