Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 264 for gather (0.16 sec)

  1. tensorflow/compiler/mlir/quantization/tensorflow/python/integration_test/quantize_model_test_base.py

      ) -> Tuple[core.Tensor, core.Tensor]:
        """Creates a basic gather model.
    
        This is intended to be used for TF1 (graph mode) tests.
    
        Args:
          input_type: type of the input index tensor for gather operation.
          use_variable_for_filter: Setting this to `True` makes the filter for the
            gather operation a `tf.Variable`.
    
        Returns:
          in_placeholder: Input tensor placeholder.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Mar 21 08:51:46 UTC 2024
    - 51.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/experimental/tac/tests/raise-target-subgraphs.mlir

    // CHECK:     %17 = "tfl.gather"(%arg13, %14) <{axis = 0 : i32, batch_dims = 0 : i32}> {tac.device = "DARWINN", tac.inference_type = "FLOAT"} : (tensor<5xi32>, tensor<?xi32>) -> tensor<?xi32>
    // CHECK:     %18 = tfl.add %arg14, %17 {fused_activation_function = "NONE", tac.device = "DARWINN", tac.inference_type = "FLOAT"} : tensor<?xi32>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:41:17 UTC 2024
    - 74.9K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/quantization/stablehlo/tests/passes/quantize_composite_functions.mlir

        return %0 : tensor<2x3x2x2xf32>
      }
    // CHECK: %[[GATHER:.+]] = "stablehlo.gather"(%[[ARG_0]], %[[ARG_1]]) {{.*}} : (tensor<3x4x2x!quant.uniform<i8:f32, {{.*}}>>, tensor<2x3x2xi32>) -> tensor<2x3x2x2x!quant.uniform<i8:f32, {{.*}}>>
    // CHECK: %[[UNIFORM_QUANTIZE_0:.+]] = stablehlo.uniform_quantize %[[GATHER]] : tensor<2x3x2x2x!quant.uniform<i8:f32, {{.*}}>>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 05:56:10 UTC 2024
    - 91.6K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/quantization/stablehlo/tests/passes/lift_quantizable_spots_as_functions.mlir

    // CHECK: return %[[XLA_CALL_MODULE:.*]] : tensor<2x3x2x2xi32>
    // CHECK: }
    
    // CHECK-LABEL: private @composite_gather_fn_1
    // CHECK: %[[GATHER:.*]] = "stablehlo.gather"(%arg0, %arg1)
    // CHECK: return %[[GATHER]] : tensor<2x3x2x2xi32>
    // CHECK: }
    
    // -----
    
    // Test that the name of composite functions are deterministic. There are 3
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 10 04:07:09 UTC 2024
    - 49.8K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/transforms/lower_tf.cc

      }
    };
    
    // Lowers ResizeNearestNeighbor to an indices computations with a gather along
    // the combined spatial dimensions. Generating the indices along the
    // width/height index could be used to gather along each of W and H dimension
    // of the input image array. To reduce to a single gather, these indices are
    // combined, so a single gather can be performed along the combined spatial
    // dimensions.
    //
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 74.9K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/quantization/stablehlo/python/integration_test/quantize_model_test.py

        module_str = self._extract_first_xla_call_module_op(
            self._output_saved_model_path
        )
        self.assertTrue(re.search('stablehlo.gather.*xi8>', module_str))
    
        # Due to other meta data, the compression is not exactly 1/4.
        self.assertLess(
            testing.get_size_ratio(
                self._output_saved_model_path, self._input_saved_model_path
            ),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 14 06:31:57 UTC 2024
    - 51.4K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/tests/decompose_resource_ops.mlir

        // CHECK-DAG: [[READVAR:%.+]] = "tf.ReadVariableOp"([[VAR]])
        // CHECK: [[GATHER:%.+]] = "tf.GatherV2"([[READVAR]], [[INDEX]], [[ZERO]]) <{batch_dims = 0 : i64}> {_xla_outside_compilation = "0"} : (tensor<*xi32>, tensor<?xi32>, tensor<i64>) -> tensor<*xi32>
        // CHECK: return [[GATHER]]
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 22 19:47:48 UTC 2024
    - 51.3K bytes
    - Viewed (0)
  8. src/runtime/traceback.go

    //		uintptr_t  Max;
    //	};
    //
    // The Context field will be zero to gather a traceback from the
    // current program execution point. In this case, the traceback
    // function will be called from C code.
    //
    // Otherwise Context will be a value previously returned by a call to
    // the context function. The traceback function should gather a stack
    // trace from that saved point in the program execution. The traceback
    Registered: Wed Jun 12 16:32:35 UTC 2024
    - Last Modified: Wed May 29 16:25:21 UTC 2024
    - 55.1K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/lite/tests/prepare-tf.mlir

      func.return %0 : tensor<7xi64>
    
      // CHECK-LABEL: scatter_nd_add
      // CHECK:  %[[GATHER:.*]] = "tf.GatherNd"(%arg0, %arg1) <{bad_indices_policy = ""}> : (tensor<7xi64>, tensor<1x1xi32>) -> tensor<1xi64>
      // CHECK:  %[[ADD:.*]] = "tf.Add"(%arg2, %[[GATHER]]) : (tensor<1xi64>, tensor<1xi64>) -> tensor<1xi64>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 29 07:26:59 UTC 2024
    - 59.8K bytes
    - Viewed (0)
  10. src/go/build/build.go

    		}
    		if !ctxt.isAbsPath(path) {
    			p.Dir = ctxt.joinPath(srcDir, path)
    		}
    		// p.Dir directory may or may not exist. Gather partial information first, check if it exists later.
    		// Determine canonical import path, if any.
    		// Exclude results where the import path would include /testdata/.
    		inTestdata := func(sub string) bool {
    Registered: Wed Jun 12 16:32:35 UTC 2024
    - Last Modified: Wed May 29 16:25:21 UTC 2024
    - 62.3K bytes
    - Viewed (0)
Back to top