- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 41 for input_idx (0.14 sec)
-
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) -
tensorflow/compiler/mlir/lite/quantization/lite/quantize_model_test.cc
const int input_idx = 0; EXPECT_THAT(subgraph->tensors[input_idx]->type, Eq(TensorType_INT8)); EXPECT_THAT(subgraph->tensors[input_idx]->name, Eq("input")); EXPECT_THAT(subgraph->tensors[input_idx]->quantization->scale, SizeIs(1)); EXPECT_THAT(subgraph->tensors[input_idx]->quantization->zero_point, SizeIs(1)); EXPECT_THAT(subgraph->tensors[input_idx]->quantization->scale[0],
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 23:15:24 UTC 2024 - 73.9K bytes - Viewed (0) -
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) -
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) -
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) -
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
indices.getType().getRank() != 1 || port[1] >= indices.getNumElements()) { return {}; } int64_t input_idx = indices.getValues<IntegerAttr>()[port[1]].getInt(); if (input_idx >= params_ty.getDimSize(0)) return {}; ValuePort req(params); req.port.push_back(input_idx); return req; } ValuePort ComputeInputComponentFor(Operation* op, ArrayRef<unsigned int> port) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0) -
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) -
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) -
tensorflow/compiler/mlir/tensorflow/transforms/mark_input_output_aliases.cc
void runOnOperation() override; }; constexpr char kAliasingAttr[] = "tf.aliasing_output"; constexpr int kUnassigned = -1; struct AliasInfo { AliasInfo() : input_index(kUnassigned), output_index(kUnassigned) {} int input_index; int output_index; }; // Idenitfy tf_device.cluster_func input-output alias pairs. // This is currently conservative, primarily handling the following base case: // ```
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 04:14:26 UTC 2024 - 7.5K bytes - Viewed (0) -
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)