Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for TensorListElementShape (0.32 sec)

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

            add_to_worklist(use.getOwner()->getResult(use.getOperandNumber()));
            continue;
          }
          // Refining the tensor list element type might change the output of
          // TensorListElementShape which is expected to be the originally assigned
          // shape to TensorList init ops. So replace it with the original element
          // shape value.
          if (auto tl_element_shape =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Jun 08 07:28:49 UTC 2024
    - 134.1K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td

      );
    
      TF_DerivedOperandTypeAttr shape_type = TF_DerivedOperandTypeAttr<1>;
      TF_DerivedResultTypeAttr element_dtype = TF_DerivedResultTypeAttr<0>;
    }
    
    def TF_TensorListElementShapeOp : TF_Op<"TensorListElementShape", [Pure]> {
      let summary = "The shape of the elements of the given list, as a tensor.";
    
      let description = [{
    input_handle: the list
      element_shape: the shape of elements of the list
      }];
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 23:24:08 UTC 2024
    - 793K bytes
    - Viewed (0)
Back to top