Searched defs:fallback_tensor (Results 1 – 4 of 4) sorted by relevance
57 FallbackTensor fallback_tensor(tensor); in TEST() local69 FallbackTensor fallback_tensor(&immutable_tensor); in TEST() local
218 auto fallback_tensor = CopyRefGpuTensorToRuntimeFallbackTensor( in ConvertDenseGpuTensorToRuntimeFallbackTensor() local
523 auto& fallback_tensor = arg->get<tensorflow::tfrt_stub::FallbackTensor>(); in KernelFallbackExecuteOpInternal() local903 const auto& fallback_tensor = arg.get(); in FallbackCopyTensorIfSmall() local
792 auto fallback_tensor = CreateRuntimeFallbackTensorFromTfTensorHandle( in RuntimeFallbackKernel() local