Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for getOutputHandle (0.52 sec)

  1. 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)
  2. 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)
  3. 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