Home
last modified time | relevance | path

Searched defs:ir_value (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/lazy/core/
H A Dtensor.cpp43 LazyTensorPtr LazyTensor::Create(Value ir_value, const BackendDevice& device) { in Create()
67 LazyTensor::LazyTensor(Value ir_value, const BackendDevice& device) in LazyTensor()
150 void LazyTensor::SetIrValue(Value ir_value) { in SetIrValue()
157 void LazyTensor::SetInPlaceIrValue(Value ir_value) { in SetInPlaceIrValue()
185 Value ir_value = CurrentIrValue(); in GetIrValue() local
H A Dlazy_graph_executor.cpp473 Value ir_value = tensor->CurrentIrValue(); in DumpBackendComputation() local
526 Value ir_value = GetIrValueForScalar(value, type, device); in GetIrValueForExpandedScalar() local
603 Value ir_value = tensors[i]->CurrentIrValue(); in CollectSyncTensors() local
660 Value ir_value = tensor->CurrentIrValue(); in ExtractIRAndPrepareTensorData() local
710 for (const auto& ir_value : ir_values) { in RunPostOrder() local
774 for (const auto& ir_value : ir_values) { in Compile() local
1071 Value ir_value = tensors[index]->CurrentIrValue(); in GetGraphHash() local
H A Ddebug_util.cpp98 torch::lazy::Value ir_value = tensor->CurrentIrValue(); in GetTensorsGraphInfo() local
108 torch::lazy::Value ir_value = tensor->CurrentIrValue(); in GetTensorsGraphInfo() local
H A Dtensor.h49 Value ir_value; member
H A Dir_dump_util.cpp251 for (auto& ir_value : values) { in ToBackend() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dhlo_to_ir_bindings.cc123 llvm::Value* CastToTypedValue(const Shape& shape, llvm::Value* ir_value, in CastToTypedValue()
143 llvm::Value* ir_value) { in GetTypedIrValue()
156 llvm::Value* ir_value, in BindHloToIrValue()
/aosp_15_r20/tools/dexter/slicer/
H A Dwriter.cc101 static void WriteEncodedValue(const ir::EncodedValue* ir_value, Section& data) { in WriteEncodedValue()