Searched defs:tensor_shapes (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | get_tuple_index_info.cc | 119 … const std::vector<ShapeVector> &tensor_shapes, bool *has_zero_tensor, in GetSliceShape()
156 const ShapeVector &data_shape, const std::vector<ShapeVector> &tensor_shapes, in ConstGetTupleIndexInfo()
|
| D | sparse_apply_centered_rms_prop.cc | 73 std::vector<ShapeVector> tensor_shapes = {var_shape, mg_shape, ms_shape, mom_shape}; in SparseApplyCenteredRMSPropInferShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | sparse_apply_momentum.cc | 81 std::map<std::string, std::shared_ptr<TensorShape>> tensor_shapes; in ValidParam() local
|
| D | sparse_apply_proximal_gradient_descent.cc | 82 std::map<std::string, std::shared_ptr<TensorShape>> tensor_shapes; in ValidParam() local
|
| D | sparse_apply_adagrad_da.cc | 122 std::map<std::string, std::shared_ptr<TensorShape>> tensor_shapes; in ValidParam() local
|
| D | sparse_apply_centered_rms_prop.cc | 98 std::map<std::string, std::shared_ptr<TensorShape>> tensor_shapes; in ValidParam() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | triplet_margin_loss_impl.cu | 23 …__ void FillAndBroadcast(const int64_t size, const size_t shape_size, const int64_t *tensor_shapes, in FillAndBroadcast()
213 … const int64_t *tensor_shapes, const int64_t *dst_shape, const size_t outer_size, in CalTripletMarginLoss()
|