Searched defs:logical_tensor (Results 1 – 4 of 4) sorted by relevance
37 VmapPhysicalView MultiBatchVmapTransform::logicalToPhysical(const Tensor& logical_tensor) { in logicalToPhysical()125 for (const auto& logical_tensor : logical_tensors) { in logicalToPhysical() local141 for (const auto& logical_tensor : logical_tensors) { in logicalToPhysical() local177 for (const auto& logical_tensor : logical_tensors) { in logicalToPhysical() local191 for (const auto& logical_tensor : logical_tensors) { in logicalToPhysical() local196 for (const auto& logical_tensor : logical_tensors) { in logicalToPhysical() local
43 VmapPhysicalView MultiBatchVmapTransform::logicalToPhysical(const Tensor& logical_tensor) { in logicalToPhysical()193 for (const auto& logical_tensor : logical_tensors) { in logicalToPhysical() local206 for (const auto& logical_tensor : logical_tensors) { in logicalToPhysical() local
84 bool isPhysicalScalarTensor(const Tensor& logical_tensor) { in isPhysicalScalarTensor()
161 desc logical_tensor() const { in logical_tensor() function