Home
last modified time | relevance | path

Searched defs:output_idx (Results 1 – 22 of 22) sorted by relevance

/external/tensorflow/tensorflow/core/ops/
Dparsing_ops.cc74 int output_idx = 0; in __anonb04296470202() local
117 int output_idx = 0; in __anonb04296470302() local
178 int output_idx = 0; in __anonb04296470402() local
273 int output_idx = 0; in __anonb04296470502() local
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dfusion_utils.cc50 string GetOutputNode(const FunctionDef& function, int output_idx) { in GetOutputNode()
56 string& GetMutableOutputNode(FunctionDef* function, int output_idx) { in GetMutableOutputNode()
255 for (int output_idx = 0; output_idx < number_of_outputs; output_idx++) in GetFunctionOutputs() local
/external/tensorflow/tensorflow/lite/java/src/main/native/
Dnativeinterpreterwrapper_jni.cc382 JNIEnv* env, jclass clazz, jlong handle, jint output_idx) { in Java_org_tensorflow_lite_NativeInterpreterWrapper_getOutputDataType()
399 JNIEnv* env, jclass clazz, jlong handle, jint output_idx) { in Java_org_tensorflow_lite_NativeInterpreterWrapper_getOutputQuantizationZeroPoint()
415 JNIEnv* env, jclass clazz, jlong handle, jint output_idx) { in Java_org_tensorflow_lite_NativeInterpreterWrapper_getOutputQuantizationScale()
/external/tensorflow/tensorflow/core/kernels/
Dmkl_aggregate_ops.cc51 size_t src1_idx = 0, src2_idx = 1, output_idx = 0; in Compute() local
Dlinalg_ops_common.cc179 for (int output_idx = 0; output_idx < context->num_outputs(); ++output_idx) { in PrepareOutputs() local
Drandom_poisson_op.cc113 for (int64 output_idx = start_output; output_idx < limit_output; in operator ()() local
Dspacetodepth_op_gpu.cu.cc90 const int output_idx = oX + (((n * block_size + bY) * block_size + bX) * in S2D_NCHW() local
Ddepthtospace_op_gpu.cu.cc90 const int output_idx = in D2S_NCHW() local
Drandom_op.cc346 for (int64 output_idx = start_output; output_idx < limit_output; in Compute() local
Dresource_variable_ops.cc90 Status CopyVariable(int output_idx, OpKernelContext* ctx, const Tensor* t) { in CopyVariable()
Dsegment_reduction_ops.cc1074 const Index output_idx = internal::SubtleMustCopy(indices_vec(i)); in Compute() local
/external/tensorflow/tensorflow/compiler/jit/
Ddeadness_analysis.cc368 Status MakeSymbolPredicate(Node* node, int output_idx, bool must_be_true, in MakeSymbolPredicate()
715 void SetPredicate(Node* n, int output_idx, Predicate* pred, in SetPredicate()
735 for (int output_idx : output_idxs) { in SetPredicate() local
1013 for (int output_idx = 0; output_idx < n->num_outputs(); output_idx++) { in HandleGeneric() local
/external/tensorflow/tensorflow/core/example/
Dexample_parser_configuration.cc155 int output_idx = dense_values_output_start + i; in ExtractExampleParserConfiguration() local
/external/tensorflow/tensorflow/core/profiler/internal/
Dtfprof_stats.cc209 int output_idx = 0; in AddGraph() local
/external/tensorflow/tensorflow/tools/graph_transforms/
Dfold_constants_lib.cc270 const int output_idx = out_edge->src_output(); in FoldConstants() local
/external/tensorflow/tensorflow/lite/tools/optimize/
Dsubgraph_quantizer.cc344 for (size_t output_idx = 0; output_idx < op->outputs.size(); ++output_idx) { in AsymmetricQuantizeInputsAndOutputs() local
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dcsv_dataset_op.cc638 size_t output_idx = out_tensors->size(); in FieldToOutput() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis.cc494 for (int64 output_idx = 0; output_idx < output_limits[spatial_dimension]; in HandleConvolution() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dfunction_optimizer.cc1898 int output_idx = 0; in InlineIndirectFunctionCall() local
Dconstant_folding.cc3041 int output_idx; in PartialConstPropThroughIdentityN() local
/external/tensorflow/tensorflow/python/framework/
Dops.py2198 def _tf_output(self, output_idx): argument
/external/mesa3d/src/gallium/drivers/r600/
Dr600_shader.c2863 int output_idx, int nc) in r600_tess_factor_read()