Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for tensorlist (0.12 sec)

  1. tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.cc

        // meaningless value -1 for non resource outputs.
        out_desc.input_index =
            it != output_to_input_alias.end() ? it->getSecond() : -1;
        // MLIR-based TF-Compiler bridge doesn't support tensorlist output yet.
        // TODO(ycao): Support tensorlist-type output.
        out_desc.is_tensor_list = false;
      }
    
      // XLA computation always uses Tuple shape.
      *xla_output_shape = xla::ShapeUtil::MakeTupleShape(shapes);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 17:24:39 UTC 2024
    - 45.3K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/xla_launch_util.cc

      }
    
      for (const XlaOutputDescription& descr : compilation_result->outputs) {
        if (descr.type == DT_VARIANT) {
          return errors::Unimplemented(
              "Support for TensorList crossing the XLA/TF boundary "
              "is not implemented");
        }
      }
    
      std::vector<TensorShape> output_tensor_shapes;
      output_tensor_shapes.reserve(ctx->num_outputs());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 16 00:36:08 UTC 2024
    - 40.4K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/passes.h

    // Materialize the MlirPassthroughOp by replacing it with the MLIR module
    // attached as an attribute.
    std::unique_ptr<OperationPass<func::FuncOp>>
    CreateMaterializePassthroughOpPass();
    
    // Replicates the TensorList init op by undoing some CSE needed for correct
    // shape assignment in shape_inference.
    std::unique_ptr<OperationPass<func::FuncOp>>
    CreateReplicateTensorListInitOpsPass();
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:18:05 UTC 2024
    - 31.8K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/quantization/tensorflow/python/py_function_lib.py

    ) -> Mapping[str, core.Tensor]:
      """Converts TensorLike values of `sample` to Tensors.
    
      Creates a copy of `sample`, where each value is converted to Tensors
      unless it is already a Tensor.
      The values are not converted in-place (i.e. `sample` is not mutated).
    
      Args:
        sample: A representative sample, which is a map of {name -> tensorlike
          value}.
    
      Returns:
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 31 05:32:11 UTC 2024
    - 27.4K bytes
    - Viewed (0)
Back to top