Home
last modified time | relevance | path

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

1234567

/external/tensorflow/tensorflow/compiler/xla/service/
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()
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()
Doptimize_input_output_buffer_alias.cc75 const ShapeIndex& output_index = index; in Build() local
/external/XNNPACK/test/
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
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
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
/external/libtextclassifier/native/utils/
Dtflite-model-executor.h125 TensorView<T> OutputView(const int output_index, in OutputView()
136 std::vector<T> Output(const int output_index, in Output()
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
/external/tensorflow/tensorflow/core/kernels/
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
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
Dreshape_util_gpu.cu.cc38 Tindex* output_index = &output_indices[sparse_index * output_rank]; in ReshapeSparseTensorKernel() local
/external/tensorflow/tensorflow/tools/graph_transforms/
Dinsert_logging.cc81 int32_t output_index; in InsertLogging() local
130 for (int output_index : node_outputs[node.name()]) { in InsertLogging() local
/external/tensorflow/tensorflow/core/kernels/data/
Doptional_ops.cc138 Status WriteOptionalWithValueToOutput(OpKernelContext* ctx, int output_index, in WriteOptionalWithValueToOutput()
150 Status WriteOptionalNoneToOutput(OpKernelContext* ctx, int output_index) { in WriteOptionalNoneToOutput()
/external/tensorflow/tensorflow/java/src/main/native/
Dgraph_operation_jni.cc83 jint output_index) { in Java_org_tensorflow_GraphOperation_shape()
133 jint output_index) { in Java_org_tensorflow_GraphOperation_dtype()
/external/python/cffi/c/
Dparse_c_type.c54 size_t output_index; member
813 int parse_c_type_from(struct _cffi_parse_info_s *info, size_t *output_index, in parse_c_type_from()
839 size_t output_index = 0; in parse_c_type() local
/external/tensorflow/tensorflow/compiler/tf2xla/
Dtf2xla.proto19 int64 output_index = 2; field
/external/rust/crates/base64/src/engine/
Dnaive.rs56 let mut output_index = 0_usize; in internal_encode() localVariable
131 let mut output_index = 0_usize; in internal_decode() localVariable
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.h235 int output_index) { in BroadcastBinaryOpOutputShapeFn()
/external/tensorflow/tensorflow/lite/c/
Dc_api_experimental.cc114 const TfLiteInterpreter* interpreter, int32_t output_index) { in TfLiteInterpreterGetOutputTensorIndex()
182 const TfLiteSignatureRunner* signature_runner, int32_t output_index) { in TfLiteSignatureRunnerGetOutputName()
/external/tensorflow/tensorflow/core/common_runtime/
Dall_to_all.cc109 int output_index = col_params_->group.members[i].rank; in Run() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Ddepth_to_space.h66 const int output_index = in DepthToSpace() local
Dspace_to_depth.h67 const int output_index = in SpaceToDepth() local
/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
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
/external/tensorflow/tensorflow/compiler/xrt/
Dxrt.proto178 int64 output_index = 2; field
188 int64 output_index = 1; field
/external/crosvm/argh_helpers/src/
Dlib.rs105 let mut output_index = 0; in flatten_subcommand() localVariable

1234567