Searched refs:npy_path (Results 1 – 3 of 3) sorted by relevance
| /external/ComputeLibrary/utils/ |
| D | GraphUtils.h | 158 …NumPyAccessor(std::string npy_path, TensorShape shape, DataType data_type, DataLayout data_layout …
612 inline std::unique_ptr<graph::ITensorAccessor> get_npy_output_accessor(const std::string &npy_path,…
615 if(npy_path.empty())
621 … return std::make_unique<NumPyAccessor>(npy_path, shape, data_type, data_layout, output_stream);
|
| D | GraphUtils.cpp | 185 NumPyAccessor::NumPyAccessor(std::string npy_path, TensorShape shape, DataType data_type, DataLayou… in NumPyAccessor() argument
186 : _npy_tensor(), _filename(std::move(npy_path)), _output_stream(output_stream) in NumPyAccessor()
|
| /external/tensorflow/tensorflow/python/debug/cli/ |
| D | analyzer_cli_test.py | 1091 npy_path = os.path.join(self._dump_root, "matmul.npy")
1093 "print_tensor", [tensor_name, "-w", npy_path],
1102 self.assertTrue(out.lines[4].startswith("Saved value to: %s (" % npy_path))
1104 self.assertAllClose([[7.0], [-2.0]], np.load(npy_path))
1317 npy_path = os.path.join(self._dump_root, "matmul_eval.npy")
1321 npy_path], screen_info={"cols": 80})
1331 self.assertTrue(out.lines[4].startswith("Saved value to: %s (" % npy_path))
1333 self.assertAllClose([[49.0, -14.0], [-14.0, 4.0]], np.load(npy_path))
|