Home
last modified time | relevance | path

Searched refs:tensor_str (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/python/training/
Dpy_checkpoint_reader.py57 def has_tensor(self, tensor_str): argument
58 return self._HasTensor(compat.as_bytes(tensor_str)) # pylint: disable=protected-access
63 def get_tensor(self, tensor_str): argument
67 self, compat.as_bytes(tensor_str))
/external/executorch/backends/apple/mps/
Dmps_preprocess.py252 tensor_str = "MPSTensor("
253 tensor_str += "datatype=" + str(mps_tensor.datatype) + ", "
254 tensor_str += "num_dims=" + str(mps_tensor.num_dims) + ", "
255 tensor_str += "dims=" + str(mps_tensor.dims) + ", "
256 tensor_str += "constant_buffer_size=" + str(mps_tensor.constant_buffer_size) + ", "
257 tensor_str += "segment_offset=" + str(mps_tensor.segment_offset)
258 tensor_str += ")"
260 return tensor_str
/external/pytorch/torch/
D_tensor_str.py306 tensor_str = ("," + "\n" * (dim - 1) + " " * (indent + 1)).join(slices)
307 return "[" + tensor_str + "]"
361 def _add_suffixes(tensor_str, suffixes, indent, force_newline): argument
362 tensor_strs = [tensor_str]
363 last_line_len = len(tensor_str) - tensor_str.rfind("\n") + 1
412 tensor_str = tensor_contents
478 tensor_str = (
542 tensor_str = (
575 tensor_str = _tensor_str(self.dequantize(), indent)
586 tensor_str = f"[\n{strs}\n]"
[all …]
/external/tensorflow/tensorflow/lite/python/analyzer_wrapper/
Dmodel_analyzer.cc69 const std::string tensor_str(const int tensor_idx, const int subgraph_idx, in tensor_str() function
105 out_stream << tensor_str(tensor_idx, subgraph_idx); in dump_tensor_detail()
161 out_stream << tensor_str(tensor_idx, subgraph_idx, model); in dump_tensor_list()
257 << tensor_str(input->tensor_index(), in dump_model_signature_defs()
265 << tensor_str(output->tensor_index(), in dump_model_signature_defs()
/external/tensorflow/tensorflow/core/runtime_fallback/util/
Dattr_util_test.cc200 std::string tensor_str = R"pb(dtype: DT_INT32 in TEST() local
209 ASSERT_EQ(ParseTensorAttrValue(tensor_str, &tensor), OkStatus()); in TEST()
216 std::string tensor_str = R"pb(foobar)pb"; in TEST() local
218 ParseTensorAttrValue(tensor_str, &tensor), in TEST()
/external/pytorch/torch/_dynamo/
Ddebug_utils.py197 tensor_str = repr(buffer)
199 tensor_str = f"torch.randn({list(buffer.shape)}, dtype={buffer.dtype})"
201 tensor_str = (
205 tensor_str = f"{tensor_str}.cuda()"
214 …tensor_str = f"torch.nn.Parameter(torch.randn({list(param.shape)}, dtype={param.dtype}{maybe_devic…
/external/tensorflow/tensorflow/python/eager/
Dtensor_test.py209 tensor_str = str(t)
210 self.assertIn("shape=%s, dtype=%s" % (t.shape, t.dtype.name), tensor_str)
211 self.assertIn(str(t), tensor_str)