| /external/pytorch/torch/csrc/inductor/aoti_eager/ |
| D | kernel_meta_info.cpp | 97 const TensorMetadata& tensor_metadata) { in operator <<() argument
98 stream << "is_symbolic_: " << tensor_metadata.is_symbolic_ << '\n'; in operator <<()
99 stream << "dtype_: " << tensor_metadata.dtype_ << '\n'; in operator <<()
100 stream << "device_: " << tensor_metadata.device_ << '\n'; in operator <<()
102 for (const auto& size : tensor_metadata.sizes_) { in operator <<()
107 for (const auto& stride : tensor_metadata.strides_) { in operator <<()
111 stream << "requires_grad_: " << tensor_metadata.requires_grad_ << '\n'; in operator <<()
112 stream << "dispatch_key_set_: " << tensor_metadata.dispatch_key_set_ << '\n'; in operator <<()
113 stream << "tensor_check_: " << tensor_metadata.tensor_check_.has_value() in operator <<()
120 TensorMetadata tensor_metadata, in ParameterMetadata() argument
[all …]
|
| D | kernel_holder.cpp | 284 auto tensor_metadata = TensorMetadata( in init_aoti_kernel_cache() local
296 tensor_metadata.build_guard(state); in init_aoti_kernel_cache()
298 return tensor_metadata; in init_aoti_kernel_cache()
337 auto tensor_metadata = build_tensor_metadata(metadata); in init_aoti_kernel_cache() local
338 test_list_metadata.push_back(tensor_metadata); in init_aoti_kernel_cache()
401 auto tensor_metadata = build_tensor_metadata(metadata); in init_aoti_kernel_cache() local
402 parameter_metadata_list.emplace_back(tensor_metadata, arg_idx); in init_aoti_kernel_cache()
|
| D | kernel_meta_info.h | 121 ParameterMetadata(TensorMetadata tensor_metadata, uint64_t input_order);
|
| /external/tflite-support/tensorflow_lite_support/cc/task/vision/utils/ |
| D | image_tensor_specs.cc | 67 const TensorMetadata& tensor_metadata) { in GetImagePropertiesIfAny() argument
68 if (tensor_metadata.content() == nullptr || in GetImagePropertiesIfAny()
69 tensor_metadata.content()->content_properties() == nullptr) { in GetImagePropertiesIfAny()
73 ContentProperties type = tensor_metadata.content()->content_properties_type(); in GetImagePropertiesIfAny()
80 tensor_metadata.name() ? tensor_metadata.name()->str() : "#0", in GetImagePropertiesIfAny()
85 return tensor_metadata.content()->content_properties_as_ImageProperties(); in GetImagePropertiesIfAny()
89 const TensorMetadata& tensor_metadata) { in GetNormalizationOptionsIfAny() argument
93 tensor_metadata, tflite::ProcessUnitOptions_NormalizationOptions)); in GetNormalizationOptionsIfAny()
|
| /external/tflite-support/tensorflow_lite_support/cc/task/vision/ |
| D | object_detector.cc | 66 const TensorMetadata& tensor_metadata) { in GetBoundingBoxProperties() argument
67 if (tensor_metadata.content() == nullptr || in GetBoundingBoxProperties()
68 tensor_metadata.content()->content_properties() == nullptr) { in GetBoundingBoxProperties()
73 tensor_metadata.name() ? tensor_metadata.name()->str() : "#0"), in GetBoundingBoxProperties()
77 ContentProperties type = tensor_metadata.content()->content_properties_type(); in GetBoundingBoxProperties()
83 tensor_metadata.name() ? tensor_metadata.name()->str() : "#0", in GetBoundingBoxProperties()
89 tensor_metadata.content()->content_properties_as_BoundingBoxProperties(); in GetBoundingBoxProperties()
127 const TensorMetadata& tensor_metadata, absl::string_view locale) { in GetLabelMapIfAny() argument
130 tensor_metadata, tflite::AssociatedFileType_TENSOR_VALUE_LABELS); in GetLabelMapIfAny()
138 tensor_metadata, tflite::AssociatedFileType_TENSOR_VALUE_LABELS, in GetLabelMapIfAny()
[all …]
|
| D | image_segmenter.cc | 114 const TensorMetadata& tensor_metadata, absl::string_view locale) { in GetLabelMapIfAny() argument
117 tensor_metadata, tflite::AssociatedFileType_TENSOR_AXIS_LABELS); in GetLabelMapIfAny()
125 tensor_metadata, tflite::AssociatedFileType_TENSOR_AXIS_LABELS, in GetLabelMapIfAny()
|
| /external/tensorflow/tensorflow/dtensor/cc/ |
| D | save_restore_util.cc | 104 for (const SavingTensorMetadata& tensor_metadata : tensor_metadatas) { in BuildSavingSpec() local
108 int index = tensor_metadata.tensor_index; in BuildSavingSpec()
109 const Layout& layout = tensor_metadata.layout; in BuildSavingSpec()
110 absl::Span<const int64_t> tensor_shape = tensor_metadata.shape; in BuildSavingSpec()
|
| /external/tflite-support/tensorflow_lite_support/metadata/cc/ |
| D | metadata_extractor.cc | 137 const tflite::TensorMetadata& tensor_metadata, in FindFirstProcessUnit() argument
140 if (tensor_metadata.process_units() == nullptr) { in FindFirstProcessUnit()
143 for (const auto process_unit : *tensor_metadata.process_units()) { in FindFirstProcessUnit()
161 const tflite::TensorMetadata& tensor_metadata, in FindFirstAssociatedFileName() argument
163 if (tensor_metadata.associated_files() == nullptr) { in FindFirstAssociatedFileName()
166 for (const auto associated_file : *tensor_metadata.associated_files()) { in FindFirstAssociatedFileName()
|
| D | metadata_extractor.h | 58 FindFirstProcessUnit(const tflite::TensorMetadata& tensor_metadata,
67 const tflite::TensorMetadata& tensor_metadata,
|
| /external/tensorflow/tensorflow/core/data/ |
| D | snapshot_utils.cc | 244 experimental::TensorMetadata* tensor_metadata = in WriteTensors() local
246 tensor.shape().AsProto(tensor_metadata->mutable_tensor_shape()); in WriteTensors()
258 tensor_metadata->set_tensor_size_bytes(size); in WriteTensors()
267 const auto& tensor_metadata = metadata.tensor_metadata(i); in WriteTensors() local
270 tensor_metadata.tensor_size_bytes()); in WriteTensors()
274 position, tensor_metadata.tensor_size_bytes()); in WriteTensors()
277 position += tensor_metadata.tensor_size_bytes(); in WriteTensors()
888 const auto& tensor_metadata = metadata->tensor_metadata(i); in SnappyUncompress() local
890 TensorShape shape(tensor_metadata.tensor_shape()); in SnappyUncompress()
898 std::make_unique<char[]>(tensor_metadata.tensor_size_bytes()); in SnappyUncompress()
[all …]
|
| /external/pytorch/torch/_inductor/ |
| D | debug.py | 620 tensor_metadata: TensorMetadata
682 x.tensor_metadata.shape,
683 x.tensor_metadata.stride,
684 x.tensor_metadata.dtype,
|
| D | codecache.py | 414 tensor_metadata: TensorMetadata
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | snapshot.proto | 46 repeated TensorMetadata tensor_metadata = 1; field
|
| /external/pytorch/torch/_dynamo/ |
| D | debug_utils.py | 669 tensor_metadata = torch._utils.get_tensor_metadata(t)
670 if tensor_metadata:
671 args.extend(f"{k}={v!r}" for k, v in tensor_metadata.items())
|
| /external/executorch/extension/flat_tensor/ |
| D | flat_tensor.fbs | 38 // 3. tensor_offset: segments[segment_offset].tensor_metadata[j].offset
|
| /external/pytorch/torch/utils/tensorboard/ |
| D | summary.py | 891 tensor_metadata = metadata.create_summary_metadata(
916 metadata=tensor_metadata,
|
| /external/pytorch/test/ |
| D | test_cuda.py | 3932 def tensor_metadata(x): function
4159 ten_metadata = [tensor_metadata(t) for t in outputs]
|