Home
last modified time | relevance | path

Searched refs:x_dim (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore/core/ops/
Dstrided_slice.cc286 …:compute_slicing_length(int64_t start_pos, int64_t end_pos, int64_t strides, int64_t x_dim) const { in compute_slicing_length()
289 if ((start_pos >= x_dim) || end_pos < -x_dim) { in compute_slicing_length()
292 if (-x_dim <= start_pos && start_pos < 0) { in compute_slicing_length()
293 start_pos += x_dim; in compute_slicing_length()
295 if (start_pos < -x_dim) { in compute_slicing_length()
298 if (-x_dim <= end_pos && end_pos < 0) { in compute_slicing_length()
299 end_pos += x_dim; in compute_slicing_length()
301 if (end_pos > x_dim) { in compute_slicing_length()
302 end_pos = x_dim; in compute_slicing_length()
311 if (start_pos < -x_dim || end_pos >= x_dim) { in compute_slicing_length()
[all …]
Dstrided_slice.h74 …int64_t compute_slicing_length(int64_t start_pos, int64_t end_pos, int64_t strides, int64_t x_dim)…
/third_party/mindspore/tests/st/probability/zhusuan/vae/
Dvae_mnist.py40 def __init__(self, x_dim, z_dim, batch_size): argument
42 self.x_dim = x_dim
50 self.fc3 = nn.Dense(500, x_dim)
82 def __init__(self, x_dim, z_dim, batch_size): argument
84 self.x_dim = x_dim
89 self.fc1 = nn.Dense(x_dim, 500)
125 x_dim = 32*32
128 generator = Generator(x_dim, z_dim, batch_size)
129 variational = Variational(x_dim, z_dim, batch_size)
/third_party/mindspore/mindspore/ops/_grad_experimental/
Dgrad_math_ops.py32 def _generate_perm(x_dim): argument
33 perm = tuple(range(x_dim - 2))
/third_party/mindspore/mindspore/ops/operations/
Darray_ops.py3001 x_dim = x_shape[i]
3004 if begin >= x_dim or end < -x_dim:
3008 if -x_dim <= begin < 0:
3009 begin += x_dim
3010 if begin < -x_dim:
3013 if -x_dim <= end < 0:
3014 end += x_dim
3015 if end > x_dim:
3017 end = x_dim
3025 if begin < -x_dim or end >= x_dim:
[all …]
/third_party/mindspore/mindspore/ops/_grad/
Dgrad_math_ops.py89 x_dim = len(x_shape)
92 x_axis = axis + x_dim