- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 16 for input_num (0.19 sec)
-
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) -
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) -
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) -
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) -
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) -
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) -
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) -
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) -
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) -
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)