Home
last modified time | relevance | path

Searched refs:output_idx (Results 1 – 25 of 37) sorted by relevance

12

/external/tensorflow/tensorflow/core/ops/
Dparsing_ops.cc34 int* output_idx) { in AddDenseOutputShapes() argument
39 c->set_output((*output_idx)++, s); in AddDenseOutputShapes()
47 int* output_idx) { in AddSparseOutputShapes() argument
54 c->set_output((*output_idx)++, c->Matrix(c->UnknownDim(), rank)); in AddSparseOutputShapes()
57 c->set_output((*output_idx)++, c->Vector(c->UnknownDim())); in AddSparseOutputShapes()
60 c->set_output((*output_idx)++, c->Vector(rank)); in AddSparseOutputShapes()
67 InferenceContext* c, int* output_idx) { in AddRaggedOutputShapes() argument
72 c->set_output((*output_idx)++, c->Vector(c->UnknownDim())); in AddRaggedOutputShapes()
76 c->set_output((*output_idx)++, c->Vector(num_splits)); in AddRaggedOutputShapes()
81 c->set_output((*output_idx)++, c->Vector(c->UnknownDim())); in AddRaggedOutputShapes()
[all …]
/external/tensorflow/tensorflow/core/kernels/
Drandom_binomial_op.cc196 for (int64 output_idx = start_output; output_idx < limit_output; in operator ()() local
199 int64 batch_idx = output_idx / samples_per_batch; in operator ()()
216 for (int64 sample_idx = output_idx % samples_per_batch; in operator ()()
217 sample_idx < samples_per_batch && output_idx < limit_output; in operator ()()
218 ++sample_idx, ++output_idx) { in operator ()()
222 for (int64 sample_idx = output_idx % samples_per_batch; in operator ()()
223 sample_idx < samples_per_batch && output_idx < limit_output; in operator ()()
224 ++sample_idx, ++output_idx) { in operator ()()
231 for (int64 sample_idx = output_idx % samples_per_batch; in operator ()()
232 sample_idx < samples_per_batch && output_idx < limit_output; in operator ()()
[all …]
Drandom_poisson_op.cc114 for (int64 output_idx = start_output; output_idx < limit_output; in operator ()() local
116 const int64 rate_idx = output_idx / num_samples; in operator ()()
135 for (int64 sample_idx = output_idx % num_samples; in operator ()()
136 sample_idx < num_samples && output_idx < limit_output; in operator ()()
137 sample_idx++, output_idx++) { in operator ()()
139 gen.Skip(kReservedSamplesPerOutput * output_idx); in operator ()()
197 for (int64 sample_idx = output_idx % num_samples; in operator ()()
198 sample_idx < num_samples && output_idx < limit_output; in operator ()()
199 sample_idx++, output_idx++) { in operator ()()
201 gen.Skip(kReservedSamplesPerOutput * output_idx); in operator ()()
Dlinalg_ops_common.cc181 for (int output_idx = 0; output_idx < context->num_outputs(); ++output_idx) { in PrepareOutputs() local
183 if (output_idx < num_outputs) { in PrepareOutputs()
186 output_matrix_shapes->at(output_idx); in PrepareOutputs()
190 output_idx, output_matrix_shape.dims())); in PrepareOutputs()
204 input_idx, output_idx, output_tensor_shape, &out)) { in PrepareOutputs()
213 output_idx, output_tensor_shape, &out)); in PrepareOutputs()
Drandom_op.cc221 for (int64 output_idx = start_output; output_idx < limit_output; in Compute() local
223 int64 alpha_idx = output_idx / num_samples; in Compute()
235 for (int64 sample_idx = output_idx % num_samples; in Compute()
236 sample_idx < num_samples && output_idx < limit_output; in Compute()
237 sample_idx++, output_idx++) { in Compute()
241 gen.Skip(kReservedSamplesPerOutput * output_idx); in Compute()
262 for (int64 sample_idx = output_idx % num_samples; in Compute()
263 sample_idx < num_samples && output_idx < limit_output; in Compute()
264 sample_idx++, output_idx++) { in Compute()
269 gen.Skip(kReservedSamplesPerOutput * output_idx); in Compute()
Dsegment_reduction_ops_impl.h837 const Index output_idx = internal::SubtleMustCopy(indices_vec(i));
838 OP_REQUIRES(context, FastBoundsCheck(output_idx, M),
839 errors::InvalidArgument("Index ", output_idx,
849 if (is_modified[output_idx]) {
851 output_flat.template chip<0>(output_idx) +=
854 output_flat.template chip<0>(output_idx) +=
859 output_flat.template chip<0>(output_idx) =
862 output_flat.template chip<0>(output_idx) =
866 is_modified[output_idx] = true;
Ddepthtospace_op_gpu.cu.cc89 const int output_idx = in D2S_NCHW() local
96 *(output_ptr + output_idx) = ldg(input_ptr + input_idx); in D2S_NCHW()
/external/tensorflow/tensorflow/python/ops/
Dcontrol_flow_util_v2.py159 op_name, _, output_idx = parts
161 op_name, output_idx = parts
165 output_idx = 0
166 tensor_name = "%s:%d" % (tensor_name, output_idx)
170 output_idx = int(output_idx)
177 tensor_name = node_def.input[output_idx]
185 output_arg_name = fdef.signature.output_arg[output_idx].name
Dcond_v2.py550 for output_idx, branch_outs in enumerate(zip(*branch_outputs)):
557 forward_graphs[branch_index].inputs[output_idx])
558 grad_graphs[branch_index].structured_outputs[output_idx] = zeros
590 for output_idx, branch_outs in enumerate(zip(*branch_outputs)):
600 branch_outputs[branch_idx][output_idx] = math_ops._as_indexed_slices(
607 output_idx=output_idx,
633 for output_idx, branch_outs in enumerate(
639 output_idx=output_idx,
/external/tensorflow/tensorflow/lite/java/src/main/native/
Dnativeinterpreterwrapper_jni.cc445 JNIEnv* env, jclass clazz, jlong handle, jint output_idx) { in Java_org_tensorflow_lite_NativeInterpreterWrapper_getOutputDataType() argument
449 const int idx = static_cast<int>(output_idx); in Java_org_tensorflow_lite_NativeInterpreterWrapper_getOutputDataType()
450 if (output_idx < 0 || output_idx >= interpreter->outputs().size()) { in Java_org_tensorflow_lite_NativeInterpreterWrapper_getOutputDataType()
452 "Failed to get %d-th output out of %d outputs", output_idx, in Java_org_tensorflow_lite_NativeInterpreterWrapper_getOutputDataType()
463 JNIEnv* env, jclass clazz, jlong handle, jint output_idx) { in Java_org_tensorflow_lite_NativeInterpreterWrapper_getOutputQuantizationZeroPoint() argument
467 const int idx = static_cast<int>(output_idx); in Java_org_tensorflow_lite_NativeInterpreterWrapper_getOutputQuantizationZeroPoint()
468 if (output_idx < 0 || output_idx >= interpreter->outputs().size()) { in Java_org_tensorflow_lite_NativeInterpreterWrapper_getOutputQuantizationZeroPoint()
470 "Failed to get %d-th output out of %d outputs", output_idx, in Java_org_tensorflow_lite_NativeInterpreterWrapper_getOutputQuantizationZeroPoint()
480 JNIEnv* env, jclass clazz, jlong handle, jint output_idx) { in Java_org_tensorflow_lite_NativeInterpreterWrapper_getOutputQuantizationScale() argument
484 const int idx = static_cast<int>(output_idx); in Java_org_tensorflow_lite_NativeInterpreterWrapper_getOutputQuantizationScale()
[all …]
/external/tensorflow/tensorflow/lite/tools/
Dverifier.cc410 for (const int output_idx : *op->outputs()) { in VerifySubGraphConsistency()
411 if (constant_tensors.find(output_idx) != constant_tensors.end()) { in VerifySubGraphConsistency()
414 output_idx, op_idx, in VerifySubGraphConsistency()
417 } else if (variable_tensors.find(output_idx) != in VerifySubGraphConsistency()
421 output_idx, op_idx, in VerifySubGraphConsistency()
424 } else if (subgraph_input_tensors.find(output_idx) != in VerifySubGraphConsistency()
428 output_idx, op_idx, in VerifySubGraphConsistency()
431 } else if (output_tensors.find(output_idx) != output_tensors.end()) { in VerifySubGraphConsistency()
435 output_idx, op_idx, in VerifySubGraphConsistency()
440 output_tensors.insert(output_idx); in VerifySubGraphConsistency()
/external/libaom/libaom/test/
Dcnn_test.cc83 for (int output_idx = 0; output_idx < num_outputs; output_idx++) { in RunMultiOutCNNTest() local
84 const float *expected_out = expected[output_idx]; in RunMultiOutCNNTest()
85 const int curr_output_chs = output_chs[output_idx]; in RunMultiOutCNNTest()
86 const int out_size = out_widths[output_idx] * out_heights[output_idx]; in RunMultiOutCNNTest()
96 << " output " << output_idx << " channel " << channel << " pixel " in RunMultiOutCNNTest()
106 EXPECT_LE(mse, tolerance) << " output " << output_idx << std::endl; in RunMultiOutCNNTest()
2483 for (int output_idx = 0; output_idx < num_outputs; output_idx++) { in TEST_F() local
2484 for (int channel = 0; channel < output_chs[output_idx]; ++channel) { in TEST_F()
2486 output_ite += output_dims[output_idx] * output_dims[output_idx]; in TEST_F()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dcsv_dataset_op.cc641 size_t output_idx = out_tensors->size(); in FieldToOutput() local
642 if (output_idx >= dataset()->out_type_.size()) { in FieldToOutput()
648 const DataType& dtype = dataset()->out_type_[output_idx]; in FieldToOutput()
652 dataset()->record_defaults_[output_idx].NumElements() != 1) { in FieldToOutput()
655 return errors::InvalidArgument("Field ", output_idx, in FieldToOutput()
665 dataset()->record_defaults_[output_idx].flat<int32>()(0); in FieldToOutput()
670 "Field ", output_idx, in FieldToOutput()
680 dataset()->record_defaults_[output_idx].flat<int64>()(0); in FieldToOutput()
685 "Field ", output_idx, in FieldToOutput()
695 dataset()->record_defaults_[output_idx].flat<float>()(0); in FieldToOutput()
[all …]
/external/tensorflow/tensorflow/compiler/xla/tools/
Ddriver.cc208 int output_idx = -1; member
279 Check(assignment.output_idx == -1, "Multiple out-parameters"); in ParseBufferAssignment()
280 assignment.output_idx = allocation_idx; in ParseBufferAssignment()
284 assignment.buffers_shape[assignment.output_idx] = shape; in ParseBufferAssignment()
302 Check(assignment.output_idx != -1, "Output not set"); in ParseBufferAssignment()
470 TupleShapeToString(assignment.buffers_shape[assignment.output_idx])); in main()
471 Display(table.AsPtr()[assignment.output_idx], in main()
472 assignment.buffers_shape[assignment.output_idx]); in main()
/external/tensorflow/tensorflow/core/common_runtime/
Dplacer_inspection_required_ops_utils.cc264 for (int output_idx = 0; output_idx < node->num_outputs(); ++output_idx) { in AddOutputIdentities() local
265 if (output_used[output_idx]) { in AddOutputIdentities()
272 TF_RETURN_IF_ERROR(add_identity(output_idx, identity_name, &identity_node)); in AddOutputIdentities()
273 VLOG(6) << "Added identity into " << node->name() << ":" << output_idx in AddOutputIdentities()
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dfusion_utils.cc50 string GetOutputNode(const FunctionDef& function, int output_idx) { in GetOutputNode() argument
52 function.signature().output_arg(output_idx).name(); in GetOutputNode()
56 string& GetMutableOutputNode(FunctionDef* function, int output_idx) { in GetMutableOutputNode() argument
58 function->signature().output_arg(output_idx).name(); in GetMutableOutputNode()
255 for (int output_idx = 0; output_idx < number_of_outputs; output_idx++) in GetFunctionOutputs() local
256 outputs.push_back(GetOutputNode(function, output_idx)); in GetFunctionOutputs()
/external/tensorflow/tensorflow/lite/micro/examples/person_detection/himax_driver/
DHM01B0_optimized.c71 const int output_idx = (output_y * w + output_x) * channels; in hm01b0_blocking_read_oneframe_scaled() local
73 buffer[output_idx + i] = value; in hm01b0_blocking_read_oneframe_scaled()
/external/tensorflow/tensorflow/lite/micro/examples/person_detection_experimental/himax_driver/
DHM01B0_optimized.c72 const int output_idx = (output_y * w + output_x) * channels; in hm01b0_blocking_read_oneframe_scaled() local
76 buffer[output_idx + i] = value - 128; in hm01b0_blocking_read_oneframe_scaled()
/external/tensorflow/tensorflow/lite/micro/kernels/
Dsvdf_test.cc191 int output_idx = 0; in ValidateSVDFGoldens() local
194 TF_LITE_MICRO_EXPECT_NEAR(expected_output[j], output_data[output_idx], in ValidateSVDFGoldens()
196 output_idx++; in ValidateSVDFGoldens()
260 int output_idx = 0; in ValidateIntegerSVDFGoldens() local
263 TF_LITE_MICRO_EXPECT_NEAR(expected_output[j], output_data[output_idx], 1); in ValidateIntegerSVDFGoldens()
264 output_idx++; in ValidateIntegerSVDFGoldens()
/external/tensorflow/tensorflow/compiler/jit/
Ddeadness_analysis.cc418 Status MakeSymbolPredicate(Node* node, int output_idx, bool must_be_true, in MakeSymbolPredicate() argument
420 TensorId tensor_id(node->name(), output_idx); in MakeSymbolPredicate()
453 Status MakeSymbolPredicate(Node* node, int output_idx, in MakeSymbolPredicate() argument
456 TensorId tensor_id(node->name(), output_idx); in MakeSymbolPredicate()
846 void SetPredicate(Node* n, int output_idx, Predicate* pred, in SetPredicate() argument
849 predicate_map_.insert({TensorId(n->name(), output_idx), pred}); in SetPredicate()
851 VLOG(4) << "For " << n->name() << ":" << output_idx << " from " in SetPredicate()
866 for (int output_idx : output_idxs) { in SetPredicate() local
867 SetPredicate(n, output_idx, pred, should_revisit); in SetPredicate()
1211 for (int output_idx = 0; output_idx < n->num_outputs(); output_idx++) { in HandleGeneric() local
[all …]
/external/tensorflow/tensorflow/core/profiler/internal/
Dtfprof_stats.cc213 int output_idx = 0; in AddGraph() local
222 DCHECK(absl::SimpleAtoi(input_parts[1], &output_idx)) in AddGraph()
223 << "Failed to parse integer: " << output_idx; in AddGraph()
231 node->AddInput(node_input, output_idx, i); in AddGraph()
/external/tensorflow/tensorflow/lite/tools/optimize/
Dquantize_model_test.cc224 const int32_t output_idx = subgraph->outputs[0]; in TEST_F() local
230 EXPECT_EQ(dequant_op->outputs[0], output_idx); in TEST_F()
234 EXPECT_EQ(subgraph->tensors[output_idx]->type, TensorType_FLOAT32); in TEST_F()
235 EXPECT_EQ(subgraph->tensors[output_idx]->name, "output"); in TEST_F()
242 if (input_idx != tensor_idx && output_idx != tensor_idx) { in TEST_F()
278 const int32_t output_idx = subgraph->outputs[0]; in TEST_F() local
286 EXPECT_EQ(quant_op_int8_uint8->outputs[0], output_idx); in TEST_F()
295 EXPECT_EQ(subgraph->tensors[output_idx]->type, TensorType_UINT8); in TEST_F()
296 EXPECT_EQ(subgraph->tensors[output_idx]->name, "output"); in TEST_F()
297 EXPECT_EQ(subgraph->tensors[output_idx]->quantization->scale.size(), 1); in TEST_F()
[all …]
Dquantize_model.cc273 const int32_t output_idx = in SetInputAndOutputTypes() local
275 if (output_idx < 0) { in SetInputAndOutputTypes()
278 subgraph->outputs[i] = output_idx; in SetInputAndOutputTypes()
584 int32_t output_idx = output.first; in QuantizeOpOutput() local
595 if (output_idx >= op->outputs.size()) { in QuantizeOpOutput()
599 output_idx, op->outputs.size(), EnumNameBuiltinOperator(op_code), in QuantizeOpOutput()
604 TensorT* output_tensor = subgraph->tensors[op->outputs[output_idx]].get(); in QuantizeOpOutput()
663 output_idx, EnumNameBuiltinOperator(op_code), subgraph_idx, op_idx); in QuantizeOpOutput()
/external/tensorflow/tensorflow/tools/graph_transforms/
Dfold_constants_lib.cc270 const int output_idx = out_edge->src_output(); in FoldConstants() local
271 TF_RETURN_IF_ERROR(ShapeHandleToTensorShape(ctx->output(output_idx), ctx, in FoldConstants()
272 &partial_shapes[output_idx])); in FoldConstants()
/external/tensorflow/tensorflow/core/example/
Dexample_parser_configuration.cc156 int output_idx = dense_values_output_start + i; in ExtractExampleParserConfiguration() local
158 strings::StrCat(node_output_prefix, output_idx); in ExtractExampleParserConfiguration()

12