Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for producer_name (0.13 sec)

  1. tensorflow/compiler/mlir/tfrt/ir/mlrt/tf_ops.td

        constant_operand_indices is sorted in ascending order.
        $operands_with_static_shape are indices of operands that are tagged with a maximum static shape.
        $producer_name is a string describing the name of the framework that added support for running this portion of the model on TPUs.
    
        Example:
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 22 21:35:32 UTC 2024
    - 6.7K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td

    target computation.
    'metadata' is a serialized TPUCompileMetadataProto describing the shapes and
    types of the inputs to the computation, as well as a mapping onto the TPU pod
    topology.
    'producer_name' is a string describing the name of the framework that add support for running this portion of the model on TPUs.
      }];
    
      let arguments = (ins
        Variadic<TF_Tensor>:$args,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 24 04:08:35 UTC 2024
    - 90.5K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tfrt/tests/ifrt/rewrite_cluster_to_ifrt_call.mlir

    func.func @serving_default(%arg0: tensor<1x3xf32>) -> (tensor<1x3xf32>) {
      %outputs  =  "tf.TPUCompilationResult"() {_tpu_compilation_status = "cluster", device = ""} : () -> tensor<!tf_type.string>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Feb 17 07:28:40 UTC 2024
    - 9K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_rewrite_pass.cc

                                                      output_types, inputs);
      auto producer_name_attr = cluster_func->getAttr("_producer_name");
      if (producer_name_attr)
        (*execute_op)->setAttr("_producer_name", producer_name_attr);
      return success();
    }
    
    // Given a `ParallelExecute`, replace it with a new `ParallelExecute`. The
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 30 21:25:12 UTC 2024
    - 29.7K bytes
    - Viewed (0)
Back to top