| /third_party/mindspore/mindspore-src/source/tests/st/ops/gpu/ |
| D | test_matrix_band_part.py | 43 def test_matrix_band_part(mode, dtype, batch_shape, rows, cols): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/cpu/ |
| D | test_matrix_band_part.py | 45 def test_matrix_band_part(mode, dtype, batch_shape, rows, cols): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/math/ |
| D | matrix_triangular_solve_gpu_kernel.cc | 100 … std::vector<int64_t> batch_shape, std::vector<int64_t> *a_broadcast_indices, in BroadcastBatchIndices()
193 std::vector<int64_t> batch_shape(output_shape.begin(), output_shape.end() - kIndex2); in Resize() local
199 std::vector<int64_t> batch_shape(output_shape.begin(), output_shape.end() - kIndex2); in Resize() local
208 std::vector<int64_t> batch_shape(output_shape.begin(), output_shape.end() - kIndex2); in Resize() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_proto/ |
| D | matrix_calculation_ops_proto.cc | 26 std::vector<int64_t> batch_shape; in CheckBatchShape() local
40 std::vector<int64_t> batch_shape; in IMPLEMT_COMMON_INFERFUNC() local
|
| D | linalg_ops_proto.cc | 168 Shape batch_shape; in CUST_IMPLEMT_INFERFUNC() local
417 Shape batch_shape; in IMPLEMT_INFERFUNC() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_proto/utils/ |
| D | linalg_ops_shape_fns.cc | 45 Shape batch_shape; in MakeBatchSquareMatrix() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | sparse_matrix_sparse_mat_mul.cc | 117 ShapeVector batch_shape = {x1_batch_shape[0]}; in FrontendSparseMatrixSparseMatMulInferShape() local
207 ShapeVector batch_shape = {x1_batch_shape[0]}; in SparseMatrixSparseMatMulInferShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | tensor_scatter_update.cc | 66 std::vector<int64_t> batch_shape; in Compute() local
|
| D | scatter_nd_update.cc | 70 std::vector<int64_t> batch_shape; in Compute() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/ |
| D | kv_cache_mgr_one_branch_fusion.cc | 94 auto batch_shape = batch_abstruct->BuildShape(); in CreateConcatNode() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/nn/probability/bijector/ |
| D | bijector.py | 124 def batch_shape(self): member in Bijector
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/nn/probability/distribution/ |
| D | distribution.py | 149 def batch_shape(self): member in Distribution
|
| /third_party/mindspore/mindspore-src/source/config/ |
| D | super_bar_config.json | 277 "batch_shape": "[]" string
|