Searched refs:CheckTensor (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/mindspore/lite/tools/benchmark/ |
| D | benchmark_base.h | 98 struct MS_API CheckTensor { struct
99 CheckTensor(const std::vector<size_t> &shape, const std::vector<float> &data,
237 CheckTensor *calibTensor = iter->second; in CompareData()
316 std::unordered_map<std::string, CheckTensor *> benchmark_data_;
|
| D | benchmark_base.cc | 197 auto *check_tensor = new (std::nothrow) CheckTensor(dims, data, strings_data); in ReadTensorData()
|
| /third_party/mindspore/mindspore/nn/probability/distribution/_utils/ |
| D | __init__.py | 20 from .utils import check_greater, check_prob, CheckTensor, CheckTuple, set_param_type
|
| D | utils.py | 308 class CheckTensor(PrimitiveWithInfer): class
314 super(CheckTensor, self).__init__("CheckTensor")
|
| /third_party/mindspore/tests/ut/python/nn/probability/distribution/ |
| D | test_utils.py | 27 cast_to_tensor, CheckTuple, CheckTensor
147 self.checktensor = CheckTensor()
|
| /third_party/mindspore/mindspore/nn/probability/bijector/ |
| D | bijector.py | 22 from ..distribution._utils.utils import CheckTensor, cast_to_tensor, raise_type_error
89 self.checktensor = CheckTensor()
|
| /third_party/mindspore/mindspore/nn/probability/distribution/ |
| D | distribution.py | 22 from ._utils.utils import CheckTuple, CheckTensor
100 self.checktensor = CheckTensor()
|