Home
last modified time | relevance | path

Searched defs:optional_tensor (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DCPUFallback.cpp187 const std::optional<at::Tensor>& optional_tensor = optional_tensorlist_args[i][idx]; in cpu_fallback() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dfold_conv_bn.cpp107 auto optional_tensor = std::optional<at::Tensor>(); in addBiasForConvIfNone() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/
H A Dconverter_gen.cc269 const auto optional_tensor = record_keeper.getClass("TFL_TensorOfOrNone"); in EmitOperandNumbers() local