Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 35 for output_dims (0.65 sec)

  1. tensorflow/compiler/mlir/quantization/tensorflow/passes/replace_cast_hacks_with_tf_xla_ops.cc

                                       const ArrayRef<int64_t> output_dims) {
      if (other_tensor_zp == 0) {
        return CreateScalarConstValue<int32_t>(builder, loc, 0);
      }
    
      auto shape = mlir::cast<ShapedType>(tensor.getType());
      SmallVector<int64_t> non_output_indices;
      for (int64_t i : llvm::seq<int64_t>(0, shape.getRank())) {
        if (absl::c_count(output_dims, i) == 0) {
          non_output_indices.push_back(i);
        }
      }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 47.1K bytes
    - Viewed (0)
  2. tensorflow/c/kernels.cc

      return tf_tensor;
    }
    
    TF_Tensor* TF_ForwardInputOrAllocateOutput(
        TF_OpKernelContext* context, const int* candidate_input_indices,
        int num_candidate_input_indices, int output_index,
        const int64_t* output_dims, int output_num_dims, int* forwarded_input,
        TF_Status* status) {
      TF_SetStatus(status, TF_OK, "");
      auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(context);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 22:53:47 UTC 2024
    - 36K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc

               << output_dims[dnums.output_batch_dimension()]);
      output_dims[dnums.output_feature_dimension()] = kernel_output_feature;
      DCOMMENT("inferrd output output_feature_dimension is "
               << output_dims[dnums.output_feature_dimension()]);
    
      std::vector<int64_t> input_spatial_dims;
      llvm::SmallVector<int64_t> window_spatial_dims;
      for (auto i = 0; i < num_spatial_dims; ++i) {
        input_spatial_dims.push_back(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Jun 08 07:28:49 UTC 2024
    - 134.1K bytes
    - Viewed (0)
  4. tensorflow/c/kernels_test.cc

        TF_Status* s = TF_NewStatus();
        int candidate_input_indices[1] = {0};
        int forwarded_input;
        int64_t output_dims[1] = {};
        TF_Tensor* output = TF_ForwardInputOrAllocateOutput(
            /*context=*/ctx, candidate_input_indices,
            /*num_candidate_input_indices=*/1,
            /*output_index=*/0, output_dims, /*output_num_dims=*/0,
            &forwarded_input, /*status=*/s);
        EXPECT_EQ(TF_OK, TF_GetCode(s));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Sep 06 19:12:29 UTC 2023
    - 50.4K bytes
    - Viewed (0)
  5. tensorflow/c/kernels.h

    // -1.
    TF_CAPI_EXPORT TF_Tensor* TF_ForwardInputOrAllocateOutput(
        TF_OpKernelContext* context, const int* candidate_input_indices,
        int num_candidate_input_indices, int output_index,
        const int64_t* output_dims, int output_num_dims, int* forwarded_input,
        TF_Status* status);
    
    // Allocates a temporary Tensor of the specified type and shape. The
    // Tensor must not be used after kernel construction is
    // complete.
    //
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jan 09 22:46:22 UTC 2024
    - 24.6K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc

        std::iota(broadcast_dims.begin() + axis, broadcast_dims.end(), axis + 1);
    
        llvm::SmallVector<int64_t, 4> output_dims =
            llvm::to_vector<4>(indices_shape);
        output_dims.insert(output_dims.begin() + axis, depth);
    
        Location loc = op.getLoc();
    
        // The iota result is the effective output shape of the computation,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 291.8K bytes
    - Viewed (0)
  7. tensorflow/c/experimental/ops/gen/generate_cpp_main.cc

            tensorflow::Flag(
                "namespace", &name_space_,
                "Compact C++ namespace, default is 'tensorflow::ops'."),
            tensorflow::Flag(
                "output_dir", &output_dir_,
                "Directory into which output files will be generated."),
            tensorflow::Flag(
                "source_dir", &source_dir_,
                "The tensorflow root directory, e.g. 'tensorflow/' for "
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Jun 21 18:51:25 UTC 2021
    - 3.4K bytes
    - Viewed (0)
  8. platforms/jvm/language-java/src/main/java/org/gradle/api/internal/tasks/compile/CleaningJavaCompiler.java

            ImmutableSet.Builder<File> outputDirs = ImmutableSet.builderWithExpectedSize(3);
            MinimalJavaCompileOptions compileOptions = spec.getCompileOptions();
            addDirectoryIfNotNull(outputDirs, spec.getDestinationDir());
            addDirectoryIfNotNull(outputDirs, compileOptions.getAnnotationProcessorGeneratedSourcesDirectory());
            addDirectoryIfNotNull(outputDirs, compileOptions.getHeaderOutputDirectory());
    Registered: Wed Jun 12 18:38:38 UTC 2024
    - Last Modified: Mon Aug 28 11:40:18 UTC 2023
    - 2.5K bytes
    - Viewed (0)
  9. tensorflow/c/experimental/ops/gen/common/path_config.cc

    #include "tensorflow/core/platform/types.h"
    
    namespace tensorflow {
    namespace generator {
    
    PathConfig::PathConfig(const string& output_dir, const string& source_dir,
                           const string& api_dir_list,
                           const std::vector<string> op_names)
        : output_path(output_dir), op_names(op_names) {
      api_dirs = str_util::Split(api_dir_list, ",", str_util::SkipEmpty());
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 09:51:28 UTC 2024
    - 2.6K bytes
    - Viewed (0)
  10. platforms/core-execution/execution/src/integTest/groovy/org/gradle/internal/execution/MutableUnitOfWorkBuilder.groovy

            Map<String, ? extends File> outputDirs,
            Collection<? extends TestFile> create,
    
            InputFingerprinter inputFingerprinter,
            ExecutionHistoryStore executionHistoryStore
        ) {
            this.inputProperties = inputProperties
            this.inputFiles = inputFiles
            this.outputFiles = outputFiles
            this.outputDirs = outputDirs
            this.create = create
    
    Registered: Wed Jun 12 18:38:38 UTC 2024
    - Last Modified: Wed Nov 22 09:41:28 UTC 2023
    - 9.2K bytes
    - Viewed (0)
Back to top