Search Options

Results per page
Sort
Preferred Languages
Advance

Results 41 - 45 of 45 for operandSegmentSizes (0.24 sec)

  1. tensorflow/compiler/mlir/tensorflow/tests/tf-ops.mlir

      func.return
    }
    
    // -----
    
    func.func @testSendTPUEmbeddingGradients(%x: tensor<512x256xf32>) {
      "tf.SendTPUEmbeddingGradients"(%x) {N = 1 : i64, NN = 0 : i64, config = "", operandSegmentSizes = array<i32: 1, 0>} : (tensor<512x256xf32>) -> ()
      func.return
    }
    
    // -----
    
    //===--------------------------------------------------------------------===//
    //  tf.BatchToSpace
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Oct 23 14:40:35 UTC 2023
    - 236.4K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/tests/canonicalize.mlir

    func.func @testXlaReduceToXlaVariadicReduceV2(%arg0: tensor<*xbf16>, %arg1: tensor<*xbf16>) -> tensor<*xbf16> {
      // CHECK: "tf.XlaVariadicReduceV2"(%arg0, %arg1) <{dimensions_to_reduce = [], operandSegmentSizes = array<i32: 1, 1>, reducer = @sum1}> {device = "/job:localhost/replica:0/task:0/device:GPU:0"} : (tensor<*xbf16>, tensor<*xbf16>) -> tensor<*xbf16>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 22:07:10 UTC 2024
    - 132.1K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/tests/tpu_rewrite.mlir

        %2:4 = "tf_device.launch"() ({
          %4 = "tf.Cast"(%arg0) {Truncate = false} : (tensor<512xi64>) -> tensor<512xi32>
          %5 = "tf.TPUCopyWithDynamicShape"(%4,  %cst) {operandSegmentSizes = array<i32: 1, 1>} : (tensor<512xi32>, tensor<i32>) -> tensor<512xi32>
          tf_device.return %5 : tensor<512xi32>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 22:03:30 UTC 2024
    - 172.9K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tf2xla/tests/legalize-tf.mlir

      // CHECK: return %[[REDUCE]]
      %0 = "tf.XlaVariadicReduceV2"(%arg0, %arg1) {_XlaHasReferenceVars = false, device = "/job:localhost/replica:0/task:0/device:XLA_GPU:0", dimensions_to_reduce = [0], operandSegmentSizes = array<i32: 1, 1>, reducer = @sum_reducer} : (tensor<2x3xcomplex<f64>>, tensor<complex<f64>>) -> tensor<3xcomplex<f64>>
      func.return %0 : tensor<3xcomplex<f64>>
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon May 06 18:46:23 UTC 2024
    - 335.5K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td

        void eraseArguments(const BitVector& erase_indices);
    
        // Gets the argument operands to the called function.
        operand_range getArgOperands() {
          // Merge `in_tensors` and `captured_tensors`. `operandSegmentSizes`
          // has two elements, so it will be merging two operand ranges (the two
          // variadic arguments). Attributes like `batch_timeout_micros` will not
          // make it to the return array.
    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