Home
last modified time | relevance | path

Searched refs:check_shape (Results 1 – 8 of 8) sorted by relevance

/third_party/boost/libs/multi_array/test/
Dconstructors.cpp24 void check_shape(const double&, std::size_t*, int*, unsigned int) in check_shape() function
28 void check_shape(const Array& A, in check_shape() function
37 check_shape(A[0], ++sizes, ++strides, num_elements / A.size()); in check_shape()
74 check_shape(A, &sizes[0], strides, num_elements); in main()
78 check_shape(B, &sizes[0], strides, num_elements); in main()
82 check_shape(C, &sizes[0], strides, num_elements); in main()
94 check_shape(A, &sizes[0], col_strides, num_elements); in main()
99 check_shape(B, &sizes[0], col_strides, num_elements); in main()
104 check_shape(C, &sizes[0], col_strides, num_elements); in main()
114 check_shape(A, &sizes[0], strides, num_elements); in main()
[all …]
/third_party/mindspore/mindspore/ops/operations/
D_embedding_cache_ops.py50 def check_shape(self, input_x_shape, indices_shape, update_shape, max_num_shape): member in UpdateCache
95 def check_shape(self, input_x_shape, max_num_shape, offset_shape): member in SubAndFilter
170 def check_shape(self, cache_table_shape, swap_cache_idx_shape, miss_value_shape): member in CacheSwapTable
302 def check_shape(self, input_x_shape, cum_sum_arr_shape, shift_idx_shape): member in PadAndShift
D_inner_ops.py642 def check_shape(self, input_shape): member in GpuConvertToDynamicShape
737 def check_shape(self, lengths_shape, maxlen_shape): member in SequenceMask
1074 def check_shape(self, indices_shape, data_shape): member in DynamicStitch
Dnn_ops.py600 def check_shape(self, input_x): member in ReLU6
3026 def check_shape(self, parameters_shape, gradient_shape, learning_rate_shape, member in SGD
4408 def check_shape(self, true_classes_shape, sampled_candidates_shape): member in ComputeAccidentalHits
6407 def check_shape(self, var_shape, accum_shape, lr_shape, l1_shape, l2_shape, member in SparseApplyProximalAdagrad
7124 def check_shape(self, var_shape, accum_shape, linear_shape, grad_shape, indices_shape): member in SparseApplyFtrl
7292 def check_shape(self, x_shape): member in Dropout
7551 def check_shape(self, inputs_shape, sequence_length_shape): member in CTCGreedyDecoder
Darray_ops.py106 def check_shape(self, x_shape, indices_shape, updates_shape): member in _ScatterOpDynamic
6025 def check_shape(self, start_shape, limit_shape, delta_shape): member in Range
6114 def check_shape(self, x_shape, mask_shape): member in MaskedSelect
Dmath_ops.py1164 def check_shape(self, x1, x2): member in MatMul
/third_party/mindspore/mindspore/ops/
Dprimitive.py387 def check_shape(self, *args): member in PrimitiveWithCheck
/third_party/mindspore/mindspore/ccsrc/debug/
Ddebug_services.cc607 std::stringstream check_shape(shape_str); in ReadTensorFromNpy() local
609 while (getline(check_shape, intermediate, ',')) { in ReadTensorFromNpy()