- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for getOutputHandle (0.17 sec)
-
tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc
auto expanded_item = rewriter.create<TF::ExpandDimsOp>( loc, expanded_item_type, item, scalar_zero); Type elem_type = getElementTypeOrSelf(item); auto handle_dtype = getElementTypeOrSelf(op.getOutputHandle().getType()) .cast<TF::VariantType>(); Type result_type = GetTensorTypeForTensorList(elem_type, handle_dtype, &rewriter);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 70.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
return false; add_to_worklist(set_item.getOutputHandle()); continue; } if (auto pop = llvm::dyn_cast<TensorListPopBackOp>(use.getOwner())) { add_to_worklist(pop.getOutputHandle()); continue; } if (auto resize = llvm::dyn_cast<TensorListResizeOp>(use.getOwner())) { add_to_worklist(resize.getOutputHandle()); continue; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0)