- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for getOutputHandle (0.24 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/tensor_list_ops_decomposition.cc
} Value size = cutil::GetR1Const({type.getShape()[0]}, builder, list.getLoc()); (*buffer_to_size)[buffer] = {size, /*fixed=*/true}; list.getOutputHandle().replaceAllUsesWith(buffer); list.erase(); return success(); } LogicalResult HandleTensorListPushBackOp( TF::TensorListPushBackOp push, llvm::SmallDenseMap<Value, SizeInfo>* buffer_to_size) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 39.2K bytes - Viewed (0) -
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)