Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for getOutputHandle (0.17 sec)

  1. 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)
  2. 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)
Back to top