| /external/pytorch/torch/csrc/jit/serialization/ |
| D | pickler.cpp | 48 void Pickler::pushIValueImpl(const IValue& ivalue) { in pushIValueImpl()
68 pushSpecializedList(ivalue, "build_intlist", [this](const IValue& ivalue) { in pushIValueImpl()
75 ivalue, "build_tensorlist", [this](const IValue& ivalue) { in pushIValueImpl()
82 ivalue, "build_doublelist", [this](const IValue& ivalue) { in pushIValueImpl()
88 pushSpecializedList(ivalue, "build_boollist", [this](const IValue& ivalue) { in pushIValueImpl()
161 void Pickler::pushDevice(const IValue& ivalue) { in pushDevice()
177 void Pickler::pushRRef(const IValue& ivalue) { in pushRRef()
198 void Pickler::pushIValue(const IValue& ivalue) { in pushIValue()
362 void Pickler::pushTensor(const IValue& ivalue) { in pushTensor()
420 void Pickler::pushLiteralTensor(const IValue& ivalue) { in pushLiteralTensor()
[all …]
|
| D | pickle.cpp | 43 const IValue& ivalue, in pickle()
52 const IValue& ivalue, in pickle()
68 std::vector<char> pickle_save(const at::IValue& ivalue) { in pickle_save()
|
| D | flatbuffer_serializer.cpp | 156 flatbuffers::Offset<mobile::serialization::IValue> ivalue) { in insertIValue()
479 const IValue& ivalue) { in dictToFB()
569 objectToFB(flatbuffers::FlatBufferBuilder& fbb, const IValue& ivalue) { in objectToFB()
605 const IValue& ivalue) { in tensorToFB()
676 const IValue& ivalue) { in storeIValueAndGetIndex()
708 iValueToFB(flatbuffers::FlatBufferBuilder& fbb, const IValue& ivalue) { in iValueToFB()
|
| /external/pytorch/torch/csrc/jit/mobile/ |
| D | flatbuffer_loader.cpp | 272 const mobile::serialization::IValue* ivalue) { in parseAndPopulate()
424 const mobile::serialization::IValue& ivalue) { in parseEnum()
445 const mobile::serialization::IValue& ivalue) { in parseBasic()
522 const mobile::serialization::IValue& ivalue) { in parseTensor()
530 const mobile::serialization::IValue& ivalue) { in parseList()
549 const mobile::serialization::IValue& ivalue) { in parseIntList()
556 const mobile::serialization::IValue& ivalue) { in parseDoubleList()
563 const mobile::serialization::IValue& ivalue) { in parseBoolList()
575 const mobile::serialization::IValue& ivalue) { in parseTuple()
588 const mobile::serialization::IValue& ivalue) { in parseDict()
[all …]
|
| /external/pytorch/torch/csrc/inductor/aoti_eager/ |
| D | kernel_holder.cpp | 27 const c10::IValue& ivalue, in unpack_tensor_ivalue()
34 const c10::IValue& ivalue, in unpack_optional_tensor_ivalue()
44 const c10::IValue& ivalue, in unpack_tensor_list_ivalue()
53 const c10::IValue& ivalue, in unpack_optional_tensor_list_ivalue()
67 const auto& ivalue = stack[idx]; in unpack_tensors() local
88 const c10::IValue& ivalue) { in is_default_value()
|
| /external/pytorch/aten/src/ATen/native/ |
| D | MathBitsFallback.h | 91 auto& ivalue = (*stack)[stack_start + i]; in fallback_impl() local
137 auto& ivalue = (*stack)[stack_start]; in fallback_impl() local
|
| /external/pytorch/aten/src/ATen/functorch/ |
| D | PlumbingHelper.h | 52 inline bool ivalueParticipatesInCurrentLevel(const IValue& ivalue) { in ivalueParticipatesInCurrentLevel()
|
| D | ADInterpreters.cpp | 144 const auto ivalue = (*stack)[idx]; in autogradBasedTransformSendToNext() local
191 auto& ivalue = (*stack)[args_front + arg_idx]; in autogradBasedTransformSendToNext() local
|
| D | BatchedFallback.cpp | 119 const auto& ivalue = arguments[idx]; in batchedTensorInplaceForLoopFallback() local
294 const auto& ivalue = arguments[idx]; in batchedTensorForLoopFallback() local
435 const auto& ivalue = arguments[idx]; in batchedNestedTensorForLoopFallback() local
|
| /external/pytorch/torch/csrc/jit/ir/ |
| D | constants.h | 55 if (auto ivalue = toIValue(v)) { in constant_as() local
|
| D | constants.cpp | 18 static bool insertableIValue(const IValue& ivalue) { in insertableIValue()
|
| /external/pytorch/torch/csrc/api/src/ |
| D | serialize.cpp | 9 std::vector<char> pickle_save(const at::IValue& ivalue) { in pickle_save()
|
| /external/pytorch/torch/csrc/api/src/serialize/ |
| D | input-archive.cpp | 22 void InputArchive::read(const std::string& key, c10::IValue& ivalue) { in read()
26 bool InputArchive::try_read(const std::string& key, c10::IValue& ivalue) { in try_read()
|
| D | output-archive.cpp | 21 void OutputArchive::write(const std::string& key, const c10::IValue& ivalue) { in write()
|
| /external/pytorch/torch/csrc/api/src/optim/ |
| D | serialize.cpp | 25 IValue ivalue; in serialize() local
|
| /external/pytorch/aten/src/ATen/ |
| D | ZeroTensorFallback.cpp | 45 auto& ivalue = (*stack)[stack_start + i]; in zeroTensorFallback() local
|
| D | LegacyBatchedFallback.cpp | 114 const auto& ivalue = arguments[idx]; in batchedTensorInplaceForLoopFallback() local
278 const auto& ivalue = arguments[idx]; in batchedTensorForLoopFallback() local
|
| D | FunctionalizeFallbackKernel.cpp | 63 const auto& ivalue = arguments[idx]; in functionalizeFallback() local
105 const auto& ivalue = returns[idx]; in functionalizeFallback() local
|
| /external/pytorch/torch/csrc/lazy/ts_backend/ |
| D | ts_eager_fallback.cpp | 186 for (const auto& ivalue : arguments) { in ltc_eager_fallback() local
225 const auto& ivalue = arguments[idx]; in ts_eager_fallback() local
|
| /external/pytorch/torch/csrc/distributed/rpc/ |
| D | types.cpp | 57 GloballyUniqueId GloballyUniqueId::fromIValue(const at::IValue& ivalue) { in fromIValue()
|
| /external/pytorch/aten/src/ATen/native/quantized/ |
| D | qconv_unpack.cpp | 182 unpack_quantized_prepacked_sizes_conv2d(const IValue& ivalue) { in unpack_quantized_prepacked_sizes_conv2d()
|
| /external/pytorch/torch/csrc/jit/runtime/static/ |
| D | memory_planner.h | 196 bool isManagedOutputTensor(const IValue& ivalue) const { in isManagedOutputTensor()
|
| /external/pytorch/test/cpp/api/ |
| D | jit.cpp | 121 torch::IValue ivalue = torch::jit::unpickle(data.data(), data.size()); in TEST() local
|
| /external/pytorch/aten/src/ATen/core/ |
| D | IListRef_inl.h | 171 const auto& ivalue = (*it).get(); in iterator_get() local
|
| /external/executorch/extension/pytree/aten_util/test/ |
| D | ivalue_util_test.cpp | 25 c10::IValue ivalue; member
|