Search Options

Results per page
Sort
Preferred Languages
Advance

Results 31 - 40 of 42 for input_idx (0.16 sec)

  1. tensorflow/compiler/mlir/lite/flatbuffer_import.cc

      if (auto* opts = options.AsCallOnceOptions()) {
        uint32_t init_idx = opts->init_subgraph_index;
        if (init_idx >= func_names.size()) {
          return errors::InvalidArgument("subgraph with index not found: ",
                                         init_idx);
        }
        auto init_attr = builder.getStringAttr(func_names.at(init_idx));
    
        return llvm::SmallVector<mlir::NamedAttribute, 4>{
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 18:21:50 UTC 2024
    - 66.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.cc

        // TODO(ycao): Support constant output.
        out_desc.is_constant = false;
        TF_RETURN_IF_ERROR(XLAShapeToTensorShape(shape, &out_desc.shape));
        // Input_index is only meaningful for resource output. Setting it to
        // meaningless value -1 for non resource outputs.
        out_desc.input_index =
            it != output_to_input_alias.end() ? it->getSecond() : -1;
        // MLIR-based TF-Compiler bridge doesn't support tensorlist output yet.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 17:24:39 UTC 2024
    - 45.3K bytes
    - Viewed (0)
  3. tensorflow/cc/gradients/array_grad.cc

      Input input = Shape(scope, op.input(0));
      Input input_min = op.input(1);
      Input input_max = op.input(2);
      int64_t axis;
      TF_RETURN_IF_ERROR(GetNodeAttr(op.node()->attrs(), "axis", &axis));
      auto qdq_v4_grad = QuantizeAndDequantizeV4Grad(
          scope, grad_inputs[0], input, input_min, input_max,
          QuantizeAndDequantizeV4Grad::Axis(axis));
      grad_outputs->push_back(qdq_v4_grad.input_backprop);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Oct 10 23:33:32 UTC 2023
    - 31.7K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc

        // output_spatial_shape[i] = ceil(input_spatial_shape[i] / strides[i])
        auto get_output_dim_for_same_padding = [](int64_t input_dim,
                                                  int64_t stride_dim) -> int64_t {
          return std::ceil(input_dim / static_cast<double>(stride_dim));
        };
        return output_height ==
                   get_output_dim_for_same_padding(input_height, stride_height) &&
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 22 09:00:19 UTC 2024
    - 99.8K bytes
    - Viewed (0)
  5. tensorflow/c/eager/c_api_test.cc

      int num_inputs = TFE_OpGetFlatInputCount(other, status);
      CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
      for (int input_index = 0; input_index < num_inputs; ++input_index) {
        TFE_TensorHandle* input = TFE_OpGetFlatInput(other, input_index, status);
        CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
        TFE_OpAddInput(ret, input, status);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Aug 03 20:50:20 UTC 2023
    - 94.6K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_driver.h

      // Checks preconditions to adjust bias scale.
      bool ShouldCheckBiasScale(Operation* op, int bias_index,
                                ArrayRef<int> input_indices,
                                QuantizedType quantized_type, int& input_index,
                                int& filter_index);
    
      // Preprocesses the constants by doing the following:
      //   - Duplicates constants if it is used by multiple ops. For example, if a
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Mar 20 11:42:17 UTC 2024
    - 16.8K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc

                                     llvm::StringRef crop_b_name) -> LogicalResult {
          int64_t input_dim = input_shape[spatial_dim_index];
          int64_t output_dim = output_shape[spatial_dim_index];
          if (!static_dims(input_dim, output_dim)) return success();
    
          int64_t input_dim_pad = input_dim * block_size;
          // If crops are unknown, the maximum output spatial dim size is input
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 146.7K bytes
    - Viewed (0)
  8. tensorflow/compiler/jit/kernels/xla_ops.cc

      std::set<int> variables_updated;
      for (const auto& resource_update : compilation_result.resource_updates) {
        if (resource_update.modified) {
          variables_updated.insert(resource_update.input_index);
        }
      }
      return GetVariableInfosFromInputs(ctx->resource_manager(), ctx->device(),
                                        inputs, variable_indices,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 22:46:36 UTC 2024
    - 41.4K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc

                     << "multiples[" << i << "] = " << m;
            }
    
            if (!ShapedType::isDynamic(input_dim) &&
                !ShapedType::isDynamic(output_dim) && output_dim != input_dim * m) {
              return op.emitOpError()
                     << "requires input.shape[" << i << "] (" << input_dim << ")"
                     << " * " << m << " to be equal to "
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 22:07:10 UTC 2024
    - 170.8K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tensorflow/translate/import_model.cc

        arg_nodes->resize(inputs.size());
        ret_nodes->resize(outputs.size());
    
        for (Node* n : GetOrderedNodes()) {
          // Handle inputs/arguments.
          auto input_it = inputs.find(n->name());
          if (input_it != inputs.end()) {
            (*arg_nodes)[input_it->second] = {n, 0};
          }
    
          // Handle outputs/returns.
          if (output_node_names.contains(n->name())) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 183.2K bytes
    - Viewed (0)
Back to top