Home
last modified time | relevance | path

Searched defs:output_index (Results 1 – 25 of 179) sorted by relevance

12345678

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_input_output_alias_config.cc29 const ShapeIndex& output_index, int64_t param_number, in SetUpAlias()
83 ShapeIndex output_index(entry.output_shape_index().begin(), in CreateFromProto() local
103 ForEachAlias([&](const ShapeIndex& output_index, const Alias& alias) { in ToString()
141 [&](const xla::ShapeIndex& output_index, std::optional<Alias> alias) { in GetAliasedOutput()
160 [&](const ShapeIndex& output_index, std::optional<Alias> aliased) { in ForEachAlias()
170 [&](const ShapeIndex& output_index, std::optional<Alias> aliased) { in ForEachAliasWithStatus()
188 const Alias& alias) -> Status { in Verify()
H A Dhlo_dataflow_analysis.cc618 ShapeIndex output_index = {0, i}; in UpdateAsyncStartValueSet() local
639 ShapeIndex output_index = {1}; in UpdateAsyncStartValueSet() local
704 ShapeIndex output_index(index.begin() + 1, index.end()); in UpdateAsyncDoneValueSet() local
1013 ShapeIndex output_index = {0}; in UpdateAllGatherStartValueSet() local
1033 const ShapeIndex& output_index = pair.first; in UpdateAllGatherDoneValueSet() local
1057 const ShapeIndex& output_index = pair.first; in UpdateAllReduceDoneValueSet() local
1739 const ShapeIndex& output_index = output_source_in_place_pair.second; in GetFusionInstructionInPlaceInputOutputPairs() local
H A Dhlo_input_output_alias_config_test.cc37 void expect_aliased(const ShapeIndex& output_index, int64_t param_number, in expect_aliased()
54 void expect_not_aliased(const ShapeIndex& output_index, int64_t param_number, in expect_not_aliased()
/aosp_15_r20/external/XNNPACK/test/
H A Ddepth-to-space-operator-tester.h180 const size_t output_index = in TestNHWCxX8() local
246 const size_t output_index = in TestNHWCxX16() local
312 const size_t output_index = in TestNHWCxX32() local
378 const size_t output_index = in TestNCHW2NHWCxX32() local
H A Dspace-to-depth-operator-tester.h179 const size_t output_index = oc in TestNHWCxX8() local
248 const size_t output_index = oc in TestNHWCxX16() local
317 const size_t output_index = oc in TestNHWCxX32() local
H A Dconstant-pad-operator-tester.h156 const size_t output_index = in TestX8() local
271 const size_t output_index = in TestX16() local
386 const size_t output_index = in TestX32() local
/aosp_15_r20/external/executorch/examples/qualcomm/qaihub_scripts/llama/runner/
H A Dio_memory.cpp132 int output_index = in prepare_io() local
183 int output_index = shard_layers_[modules_.size() - 1] * 2 * in prepare_io() local
294 int output_index = in prepare_io() local
383 int output_index = shard_layers_[modules_.size() - 1] * 2 * in prepare_io() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dsegment_reduction_ops_gpu.cu.h115 const Index output_index = in DEFINE_REDUCE_UPDATE_OP_FOR() local
135 const Index output_index = in DEFINE_REDUCE_UPDATE_OP_FOR() local
187 const int64_t output_index = in UnsortedSegmentCustomKernel() local
H A Dreduce_join_op.cc50 int64_t output_index, const gtl::InlinedVector<int32, 8>& dim_list, in LinearSubIndexToFullIndex()
165 for (int64_t output_index = 0; output_index < output_shape.num_elements(); in Compute() local
H A Ddecode_proto_op.cc227 int output_index = -1; member
696 const int output_index = output_indices[i]; in DecodeProtoOp() local
925 for (int output_index = 0; output_index < field_count; ++output_index) { in AccumulateFields() local
/aosp_15_r20/external/libtextclassifier/native/utils/
H A Dtflite-model-executor.h125 TensorView<T> OutputView(const int output_index, in OutputView()
136 std::vector<T> Output(const int output_index, in Output()
H A Dtflite-model-executor.cc384 const int output_index, const tflite::Interpreter* interpreter) const { in Output() argument
397 const int output_index, const tflite::Interpreter* interpreter) const { in Output() argument
/aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/
H A Dinsert_logging.cc81 int32_t output_index; in InsertLogging() local
130 for (int output_index : node_outputs[node.name()]) { in InsertLogging() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Doptional_ops.cc138 Status WriteOptionalWithValueToOutput(OpKernelContext* ctx, int output_index, in WriteOptionalWithValueToOutput()
150 Status WriteOptionalNoneToOutput(OpKernelContext* ctx, int output_index) { in WriteOptionalNoneToOutput()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DUpSample.h335 int64_t output_index, in nearest_idx()
356 int64_t output_index, in nearest_exact_idx()
452 int64_t output_index, in compute_source_index_and_lambda()
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/
H A Ddtensor_device_util.cc231 StatusOr<Layout> GetLayoutThroughIdentityOps(Node* op, int output_index) { in GetLayoutThroughIdentityOps()
542 for (int output_index = 0; output_index < output_layouts.size(); in CacheKeyForGraph() local
805 for (int output_index = 0; output_index < op_node->num_outputs(); in PrepareGraphForMlir() local
910 const int output_index = retval_edge->src_output(); in IdentifyAllFunctionsToExecute() local
/aosp_15_r20/external/executorch/backends/mediatek/runtime/
H A DNeuronBackend.cpp133 auto output_index = o - inputCount; in execute() local
174 auto output_index = o - inputCount; in HintNeuronBackend() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/python/
H A Dop_hint.py494 def aggregate_and_return_name_for_output(self, fused_op_name, output_index, argument
614 def aggregate_and_return_name_for_output(self, fused_op_name, output_index, argument
898 def _tensorflow_output_name(tensor_name, output_index): argument
/aosp_15_r20/external/tensorflow/tensorflow/java/src/main/native/
H A Dgraph_operation_jni.cc83 jint output_index) { in Java_org_tensorflow_GraphOperation_shape()
133 jint output_index) { in Java_org_tensorflow_GraphOperation_dtype()
/aosp_15_r20/external/rust/android-crates-io/crates/base64/src/engine/
Dnaive.rs55 let mut output_index = 0_usize; in internal_encode() localVariable
129 let output_index = chunk_index * 3; in internal_decode() localVariable
/aosp_15_r20/external/executorch/examples/qualcomm/executor_runner/
H A Dqnn_executor_runner.cpp281 for (int output_index = 0; output_index < method->outputs_size(); in main() local
418 for (size_t output_index = 0; output_index < method->outputs_size(); in main() local
/aosp_15_r20/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/
H A Dragged_tensor_to_tensor_tflite.cc451 const std::vector<int>& output_index, in SetOutputT()
526 const std::vector<int>& output_index, in SetOutput()
655 std::vector<int> output_index, new_output_index; in Eval() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/c/
H A Dc_api_experimental.cc114 const TfLiteInterpreter* interpreter, int32_t output_index) { in TfLiteInterpreterGetOutputTensorIndex()
182 const TfLiteSignatureRunner* signature_runner, int32_t output_index) { in TfLiteSignatureRunnerGetOutputName()
/aosp_15_r20/external/mesa3d/src/gallium/targets/teflon/
H A Dtest_executor.cpp62 unsigned output_index; in patch_conv2d() local
212 unsigned output_index = subgraph->operators[2]->outputs.data()[0]; in add_generate_model() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
H A Dbuffer_reuse_pass.cc75 int32_t output_index = kIndexAmbiguous; in find_output_indices() local
250 if (auto output_index = analysis.get_output_index(op)) { in runOnOperation() local

12345678