Searched defs:check_shapes (Results 1 – 9 of 9) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | dynamic_gru_v2_grad.cc | 108 std::map<std::string, BaseShapePtr> check_shapes = { in DynamicGRUV2GradCheckShapeValue() local
138 std::vector<ShapeVector> check_shapes = {x_shape, winput_shape, whidden_shape, y_shape}; in DynamicGRUV2GradInferShape() local
|
| D | apply_keras_momentum.cc | 71 std::vector<abstract::BaseShapePtr> check_shapes = {var_shape, accum_shape, grad_shape}; in ApplyKerasMomentumInferShape() local
|
| D | sparse_apply_proximal_adagrad.cc | 33 std::vector<ShapeVector> check_shapes = {var_shape, accum_shape, grad_shape, indices_shape}; in SparseApplyProximalAdagradCheckTensorShapeAndSize() local
|
| D | sparse_apply_r_m_s_prop.cc | 59 std::vector<ShapeVector> check_shapes = {ms_shape, mom_shape, grad_shape, var_shape}; in SparseApplyRMSPropInferShape() local
|
| D | edit_distance.cc | 104 …std::vector<ShapeVector> check_shapes = {hypothesis_indices_shape, hypothesis_values_shape, hypoth… in EditDistanceInferShape() local
|
| D | sparse_matrix_mat_mul.cc | 49 std::vector<ShapeVector> check_shapes = {x1_dense_shape, x1_batch_pointer, x2_dense_shape}; in SparseMatrixMatMulCheckShape() local
|
| D | sample_distorted_bounding_box_v2.cc | 70 …std::vector<ShapeVector> check_shapes = {image_size_shape, bboxes_shape, min_object_covered_shape}; in SampleDistortedBoundingBoxV2InferShape() local
|
| D | sparse_matrix_add.cc | 108 std::vector<ShapeVector> check_shapes = {a_indices, a_values, b_indices, b_values}; in SparseMatrixAddInferShape() local
|
| D | matrix_diag_v3.cc | 139 std::vector<ShapeVector> check_shapes = {row_shape, col_shape, padding_shape}; in MatrixDiagV3InferShape() local
|