Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 8 of 8 for EmptyTensorListOp (0.22 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/replicate_tensor_list_init_ops_pass.cc

        getOperation().walk([](Operation* op) {
          if (auto tl_reserve = dyn_cast<TensorListReserveOp>(op)) {
            ReplicateTensorListForUses(tl_reserve);
          }
          if (auto tl_empty = dyn_cast<EmptyTensorListOp>(op)) {
            ReplicateTensorListForUses(tl_empty);
          }
        });
      }
    };
    }  // namespace
    
    std::unique_ptr<OperationPass<func::FuncOp>>
    CreateReplicateTensorListInitOpsPass() {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sun Jan 22 17:28:34 UTC 2023
    - 2.7K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/transforms/legalize_tensorlist.cc

        tflite::TensorType tflite_type =
            tflite::ConvertTypeToTensorType(reserve.getElementDtype());
    
        return CreateListReserveOptions(context, tflite_type);
      }
      if (auto empty = llvm::dyn_cast_or_null<mlir::TF::EmptyTensorListOp>(op)) {
        tflite::TensorType tflite_type =
            tflite::ConvertTypeToTensorType(empty.getElementDtype());
    
        return CreateListReserveOptions(context, tflite_type);
      }
      return {};
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 07 23:04:40 UTC 2024
    - 10.6K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/tensor_list_ops_decomposition.cc

      if (!tensor_type.hasStaticShape()) return failure();
      for (auto d : tensor_type.getShape()) shape->push_back(d);
      return success();
    }
    
    LogicalResult HandleEmptyTensorListOp(
        TF::EmptyTensorListOp list,
        llvm::SmallDenseMap<Value, SizeInfo>* buffer_to_size) {
      Value buffer;
      OpBuilder builder(list);
      llvm::SmallVector<int64_t, 8> element_shape;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 39.2K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/quantization/tensorflow/passes/duplicate_shape_determining_constants.cc

          CompileTimeConstantOperand<TF::DepthwiseConv2dNativeBackpropInputOp, 0>,
          CompileTimeConstantOperand<TF::EmptyOp, 0>,  // $shape
          // $element_shape, $max_num_elements
          CompileTimeConstantOperand<TF::EmptyTensorListOp, 0, 1>,
          CompileTimeConstantOperand<TF::ExpandDimsOp, 1>,   // $dim
          CompileTimeConstantOperand<TF::FillOp, 0>,         // $dims
          CompileTimeConstantOperand<TF::GatherV2Op, 2>,     // $axis
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Mar 22 05:52:39 UTC 2024
    - 17.5K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc

    // any restrictions on the number of elements we can support. So this may
    // have a different behavior compared to TensorFlow in case of errors.
    struct ConvertEmptyTensorList
        : public ConvertTensorListInitOp<TF::EmptyTensorListOp> {
      explicit ConvertEmptyTensorList(MLIRContext *context,
                                      bool allow_tensorlist_pass_through,
                                      bool default_to_single_batch)
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 70.7K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tf2xla/internal/passes/mark_ops_for_outside_compilation.cc

          GET_OPERATION_NAME(mlir::TF::TensorArrayGatherV3Op),
          GET_OPERATION_NAME(mlir::TF::TensorArrayScatterV3Op),
          // Tensor List Ops.
          GET_OPERATION_NAME(mlir::TF::EmptyTensorListOp),
          GET_OPERATION_NAME(mlir::TF::TensorListReserveOp),
          GET_OPERATION_NAME(mlir::TF::TensorListFromTensorOp),
          GET_OPERATION_NAME(mlir::TF::TensorListPushBackOp),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 21.4K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc

      }
    
      return {};
    }
    
    //===----------------------------------------------------------------------===//
    // EmptyTensorListOp
    //===----------------------------------------------------------------------===//
    
    LogicalResult EmptyTensorListOp::verify() {
      EmptyTensorListOp op = *this;
      // This is required to populate derived attributes during export in a
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 146.7K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc

    bool IsTensorListInitOp(Operation* op) {
      return isa<TensorListReserveOp>(op) || isa<EmptyTensorListOp>(op) ||
             isa<TensorListFromTensorOp>(op);
    }
    
    // Returns the `element_shape` operand of the ops that create a TensorList.
    Value GetElementShapeOperand(Operation* op) {
      if (auto empty_tl = dyn_cast<EmptyTensorListOp>(op))
        return empty_tl.getElementShape();
      if (auto tl_reserve = dyn_cast<TensorListReserveOp>(op))
    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