Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 23 for input_idx (0.45 sec)

  1. tensorflow/compiler/jit/node_matchers.cc

                        << " inputs but node has " << node->num_inputs();
            }
            return false;
          }
    
          for (int input_idx = 0, e = input_matchers->size(); input_idx < e;
               input_idx++) {
            if (!MatchAndExplainInput(node, input_idx, listener)) {
              return false;
            }
          }
        }
    
        std::vector<const Node*> control_deps;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Jun 03 16:15:20 UTC 2022
    - 16.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/quantization/tools/tflite_op_coverage_spec_getters_gen.cc

        // the first argument is the input activation.
        int input_idx = 0;
        for (int i = 0; i < args_in_dag->getNumArgs(); i++) {
          if (args_in_dag->getArgName(i)->getAsString() == "\"input\"")
            input_idx = i;
        }
        if (CheckTypeConstraints(args_in_dag->getArg(input_idx), required_types,
                                 per_axis)) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Sep 06 19:12:29 UTC 2023
    - 12.7K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/build_xla_ops_pass.cc

                           cluster_info.resource_inputs.size());
      int input_idx = 0;
      for (const Output& o : cluster_info.non_constant_inputs) {
        if (debugging_opts.check_input_numerics && DataTypeIsFloating(o.type())) {
          ops::CheckNumerics check_numerics_op(
              s.WithOpName("check_input_", input_idx), o,
              absl::StrCat("CheckNumerics failed for input ", input_idx, "(",
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 24.3K bytes
    - Viewed (0)
  4. tensorflow/compiler/jit/xla_launch_util.cc

              compilation_result, i));
        } else if (type == DT_RESOURCE) {
          int input_index =
              compilation_result->outputs[i].input_index - missing_ctx_input_prefix;
          TF_RET_CHECK(input_index >= 0 && input_index < ctx->num_inputs())
              << "Invalid input for outputs " << i << ": " << input_index;
          ctx->set_output(i, ctx->input(input_index));
        } else {
          TF_ASSIGN_OR_RETURN(
              Tensor output_tensor,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 16 00:36:08 UTC 2024
    - 40.4K bytes
    - Viewed (0)
  5. tensorflow/c/kernels_experimental.cc

                                                       int32_t input_index,
                                                       int32_t output_index) {
      auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx);
      if (cc_ctx->input_dtype(input_index) != tensorflow::DT_RESOURCE) {
        cc_ctx->forward_ref_input_to_ref_output(input_index, output_index);
      }
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 23 06:12:29 UTC 2024
    - 30.9K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/quantization/tensorflow/passes/quantized_function_library_tf_drq.mlir

        // represented.
        %input_1d = "tf.Reshape"(%input, %shape) : (tensor<*xf32>, tensor<1xi32>) -> tensor<?xf32>
        %r_max_without_zero = "tf.Max"(%input_1d, %dim) { keep_dims = true }: (tensor<?xf32>, tensor<1xi64>) -> tensor<1xf32>
        %r_max = "tf.Maximum"(%zero, %r_max_without_zero) : (tensor<1xf32>, tensor<1xf32>) -> tensor<1xf32>
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Mar 03 15:43:38 UTC 2023
    - 12.2K bytes
    - Viewed (0)
  7. tensorflow/c/eager/parallel_device/parallel_device_lib.cc

        TFE_OpReset(op_.get(), operation_name, device_.c_str(), status);
        if (TF_GetCode(status) != TF_OK) return;
      }
      TFE_OpAddAttrs(op_.get(), attributes);
      for (int input_index = 0; input_index < inputs.size(); ++input_index) {
        TFE_OpAddInput(op_.get(), inputs[input_index], status);
        if (TF_GetCode(status) != TF_OK) return;
      }
      std::vector<TFE_TensorHandle*> unwrapped_results(expected_max_outputs);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Feb 09 07:47:20 UTC 2024
    - 25.4K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/quantization/tensorflow/cc/constant_fold_test.cc

            %weight = "tf.Const"() {value = dense<1> : tensor<1024x24x24x3xi8>} : () -> tensor<1024x24x24x3xi8>
            %input_i32 = "tf.Cast"(%weight) : (tensor<1024x24x24x3xi8>) -> tensor<1024x24x24x3xi32>
            %output = "tf.Sub"(%input_i32, %zp) : (tensor<1024x24x24x3xi32>, tensor<i32>) -> tensor<1024x24x24x3xi32>
            %cast = "tf.Cast"(%output) : (tensor<1024x24x24x3xi32>) -> tensor<1024x24x24x3xf32>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 04 07:19:09 UTC 2024
    - 10.1K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/lite/transforms/prepare_quantize_helper.h

                                    const_op->getResult(0));
        auto dq = rewriter.create<DQ>(const_op->getLoc(), expressed_type, q);
        op.setOperand(input_index, dq.getResult());
        return success();
      }
    
      LogicalResult replaceStatsOp(
          SourceOp op, quantfork::StatisticsOp stats_op, int input_index,
          const operator_property::TensorProperty& tensor_property,
          PatternRewriter& rewriter) const {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 03 18:01:23 UTC 2024
    - 28K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tensorflow/utils/tf_xla_mlir_translate.cc

               << absl::StrJoin(xla_output_description.shape.dim_sizes(), ", ")
               << ')';
        if (xla_output_description.input_index >= 0)
          output << " input_index=" << xla_output_description.input_index;
        if (xla_output_description.is_constant) output << " constant";
        if (xla_output_description.is_tensor_list) output << " tensor_list";
        output << '\n';
      }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 18.8K bytes
    - Viewed (0)
Back to top