Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 25 for tensor2 (0.35 sec)

  1. tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc

    //
    // Conditions for `tfl.fully_connected` conversion:
    //   * Input tensors are per-tensor uniform quantized (i8->f32)
    //     tensors.
    //   * The filter tensor is constant a per-tensor uniform quantized (i8->f32)
    //     tensor. The quantization dimension should be 1 (the non-contracting
    //     dimension).
    //   * Output tensors are per-tensor uniform quantized (i8->f32) or
    //     per-channel uniform quantized (i32->f32) tensors.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 22 09:00:19 UTC 2024
    - 99.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/ir/tfl_ops.td

        SameOperandsAndResultsScale]> {
      let summary = "Packs a list of tensors along a dimension into one tensor";
    
      let description = [{
        Packs a list of `values_count` rank-`R` tensors into one rank-`(R+1)`
        tensor.
    
        Packs the `values_count` tensors in `values` into a tensor with rank one
        higher than each tensor in `values`, by packing them along the `axis`
        dimension.
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 19:09:08 UTC 2024
    - 186K bytes
    - Viewed (0)
  3. tensorflow/c/c_api.cc

        // Input tensors
        const std::vector<std::pair<string, Tensor>>& input_pairs,
        // Output tensors
        const std::vector<string>& output_tensor_names, TF_Tensor** c_outputs,
        // Target nodes
        const std::vector<string>& target_oper_names, TF_Buffer* run_metadata,
        TF_Status* status) {
      const int noutputs = output_tensor_names.size();
      std::vector<Tensor> outputs(noutputs);
      Status result;
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 15 03:35:10 UTC 2024
    - 102.3K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.td

          }
          return %graph#0, %graph#1, %graph#2, %graph#3 : !tf_res, !tf_res, tensor<f32>, tensor<f32>
        }
        ```
    
        After:
    
        ```mlir
        func @while_body(%arg0: !tf_res, %arg1: !tf_res, %arg2: tensor<f32>, %arg3: tensor<f32>, %chain_0: tensor<i32>, %chain_1: tensor<i32>) -> (!tf_res, !tf_res, tensor<f32>, tensor<f32>, tensor<i32>, tensor<i32>) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:18:05 UTC 2024
    - 99.6K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/stablehlo/tests/uniform-quantized-stablehlo-to-tfl.mlir

    func.func @uniform_dequantize_op(%arg: tensor<2x2x!quant.uniform<i8:f32, 1.000000e+0:8>>) -> tensor<2x2xf32> {
      %0 = stablehlo.uniform_dequantize %arg : (tensor<2x2x!quant.uniform<i8:f32, 1.000000e+0:8>>) -> tensor<2x2xf32>
      return %0 : tensor<2x2xf32>
    }
    // CHECK-LABEL: uniform_dequantize_op
    // CHECK: %[[DEQUANT:.+]] = "tfl.dequantize"({{.*}}) : (tensor<2x2x!quant.uniform<i8:f32, 1.000000e+00:8>>) -> tensor<2x2xf32>
    // CHECK: return %[[DEQUANT]]
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 14 17:10:32 UTC 2024
    - 106.2K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tensorflow/tests/shape_inference.mlir

        // CHECK: %[[ADD:.*]] = "tf.Add"(%[[MUL]], %[[MUL]]) : (tensor<1xf32>, tensor<1xf32>) -> tensor<1xf32>
        // CHECK: return %[[ADD]] : tensor<1xf32>
        %0 = "tf.Mul"(%arg0, %arg0) : (tensor<1xf32>, tensor<1xf32>) -> tensor<*xf32>
        %1 = "tf.Add"(%0, %0) : (tensor<*xf32>, tensor<*xf32>) -> tensor<*xf32>
        func.return %1 : tensor<*xf32>
      }
    
      // CHECK-LABEL: func @simple_chain_with_broadcast
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jan 23 17:24:10 UTC 2024
    - 167.4K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/lite/flatbuffer_export.cc

          return tflite::CreateBuffer(builder_, buffer_data);
        }
      }
    
      tensorflow::Tensor tensor;
      auto status = tensorflow::ConvertToTensor(attr, &tensor);
      if (!status.ok()) {
        inst->emitError(
            Twine("failed to convert value attribute to tensor with error: " +
                  status.ToString()));
        return std::nullopt;
      }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:41:49 UTC 2024
    - 164.5K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tensorflow/tests/tf-ops.mlir

    func.func @testBroadcastGradientArgs(%s0: tensor<4xi32>, %s1: tensor<4xi32>) -> (tensor<1xi32>, tensor<0xi32>) {
      %r0, %r1 = "tf.BroadcastGradientArgs"(%s0, %s1) : (tensor<4xi32>, tensor<4xi32>) -> (tensor<1xi32>, tensor<0xi32>)
      func.return %r0, %r1 : tensor<1xi32>, tensor<0xi32>
    }
    
    // -----
    
    func.func @testBroadcastGradientArgsIncompatibleInputType(%s0: tensor<4xi32>, %s1: tensor<4xi64>) -> (tensor<1xi32>, tensor<0xi32>) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Oct 23 14:40:35 UTC 2023
    - 236.4K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tensorflow/translate/import_model.cc

          return errors::FailedPrecondition(
              "Feed output tensor must be a data output '", tensor.ToString(), "'");
    
        auto& node = feeds_by_node[tensor.node()];
        if (!node.insert({tensor.index(), &input}).second)
          return errors::FailedPrecondition(
              "Multiple feeds for the same output tensor '", tensor.ToString(),
              "'");
      }
    
      return feeds_by_node;
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 183.2K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc

        return op.emitOpError("requires min to be a 1d float tensor");
    
      auto max = GetRankedTensorTypeForOperand(op.getMax());
      if (max && !IsOfRankedFloatTensorType(max, 1))
        return op.emitOpError("requires max to be a 1d float tensor");
    
      Value inputs = op.getInputs();
      if (!HasRankAtLeast(inputs, 1))
        return op.emitError("requires inputs to be at least 1d float tensor");
    
      int64_t num_bits = op.getNumBits();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 146.7K bytes
    - Viewed (0)
Back to top