Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 16 for input_num (0.19 sec)

  1. tensorflow/compiler/jit/get_compiler_ir.cc

      // XLA compiler arguments.
      for (int64_t input_num = offset; input_num < input_arg_size; ++input_num) {
        const Tensor* input = inputs[input_num];
    
        XlaCompiler::Argument arg;
        if (variable_info_lookup.count(input_num)) {
          // Handles tf.resource variables.
          TF_RET_CHECK(input->dtype() == DT_RESOURCE);
          const VariableInfo& variable = *variable_info_lookup[input_num];
          arg.kind = XlaCompiler::Argument::kResource;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 06:59:07 UTC 2024
    - 19K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/xla_launch_util.cc

      TF_CHECK_OK(CreateVariableInfoLookup(variable_args, variable_info_lookup));
      for (int64_t input_num = 0; input_num < inputs.size(); ++input_num) {
        const Tensor* input = inputs[input_num];
    
        XlaCompiler::Argument& arg = out[input_num];
        if (variable_info_lookup.count(input_num) && device != nullptr) {
          // Handles resource variables.
          TF_RET_CHECK(input->dtype() == DT_RESOURCE);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 16 00:36:08 UTC 2024
    - 40.4K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/flatbuffer_import.cc

      OperationState op_state(loc, op_name);
    
      for (auto input_num : op.inputs) {
        if (input_num == -1) {
          assert(optional_arg_marker != nullptr);
          op_state.addOperands({optional_arg_marker});
        } else {
          op_state.addOperands({vals_map.at(input_num)});
        }
      }
    
      for (auto output_num : op.outputs) {
        auto& tensor = *tensors.at(output_num);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 18:21:50 UTC 2024
    - 66.8K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/quantization/tensorflow/cc/convert_asset_args.cc

        if (!asset_op) continue;
    
        const StringRef input_name = input_names[arg_idx];
        ConvertMainArgAttrs(main_func_op, arg_idx, /*index_path=*/input_name);
    
        // This assumes that the final tensor name in `GraphDef` is equal to
        // `input_name`.
        asset_file_defs.emplace_back(CreateAssetFileDef(
            asset_op.getFilenameAttr(), /*tensor_name=*/input_name));
      }
    
      return asset_file_defs;
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 5.4K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/quantization/tensorflow/python/save_model.py

          for idx, input_name in enumerate(node.input):
            if input_name in output_renaming_map:
              node.input[idx] = output_renaming_map[input_name]
          # Update the control inputs referring to the pre-renaming node.
          updating_inputs = []
          for input_name in reversed(node.input):
            if input_name.startswith('^') and input_name[1:] in output_renaming_map:
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 01:09:50 UTC 2024
    - 12.3K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/lite/sparsity/sparsify_model_test.cc

      std::string expected_key = tflite::optimize::kTfLiteReducedPrecisionKey;
      std::string expected_value = "test_data";
    
      // Load input model
      auto input_fbm = tflite::FlatBufferModel::BuildFromFile(
          "tensorflow/lite/testdata/sparse_tensor.bin");
      tflite::ModelT input_model;
      input_fbm->GetModel()->UnPackTo(&input_model);
    
      // Populate input metadata
      auto model_metadata_buffer = std::make_unique<tflite::BufferT>();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Jun 10 20:16:40 UTC 2024
    - 2.9K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_main_function.cc

            for (StringRef input_name : fn_input_names) {
              if (input_name_set.contains(input_name)) {
                // Found a duplicated name, all input names will be prefixed by
                // their corresponding function names.
                need_prefix_for_input_name = true;
              }
              input_name_set.insert(input_name);
              fn_input_name_vec.push_back(function_name);
            }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 16.5K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/quantization/tensorflow/python/representative_dataset.py

                  ' for representative samples.'
              )
    
            sample = _RepresentativeDataSample()
            for input_name, input_value in repr_sample.items():
              sample.tensor_proto_inputs[input_name].CopyFrom(
                  tensor_util.make_tensor_proto(input_value)
              )
    
            writer.write(sample.SerializeToString())
    
        logging.info(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Mar 22 22:55:22 UTC 2024
    - 14.2K bytes
    - Viewed (0)
  9. tensorflow/c/kernels_experimental.h

    // `TF_MaybeLockVariableInputMutexesInOrder` API
    TF_CAPI_EXPORT extern void TF_ReleaseVariableInputLockHolder(
        TF_VariableInputLockHolder* lockHolder);
    
    // Allows plugin to get TF_Tensor when passed its input_name
    TF_CAPI_EXPORT extern void TF_GetInputByName(TF_OpKernelContext* ctx,
                                                 const char* inputName,
                                                 TF_Tensor** tensor,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Aug 07 14:44:39 UTC 2023
    - 9.4K bytes
    - Viewed (0)
  10. tensorflow/c/eager/c_api.cc

    }
    
    TF_CAPI_EXPORT extern int TFE_OpGetInputLength(TFE_Op* op,
                                                   const char* input_name,
                                                   TF_Status* status) {
      int ret = -1;
      status->status = tensorflow::unwrap(op)->InputLength(input_name, &ret);
      return ret;
    }
    
    TF_CAPI_EXPORT extern int TFE_OpGetOutputLength(TFE_Op* op,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 08:11:23 UTC 2024
    - 44K bytes
    - Viewed (0)
Back to top