Home
last modified time | relevance | path

Searched refs:get_output_index (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
Dbuffer_reuse_pass.cc60 Optional<int32_t> get_output_index(memref::AllocOp op) { in get_output_index() function in mlir::kernel_gen::transforms::__anon3a7b9e060111::BufferReuseAnalysis
250 if (auto output_index = analysis.get_output_index(op)) { in runOnOperation()
/external/executorch/runtime/executor/
Dmethod.cpp955 auto& output = mutable_value(get_output_index(output_idx)); in set_output_data_ptr()
1001 output_evalues[i] = values_[get_output_index(i)]; in get_outputs()
1356 size_t Method::get_output_index(size_t i) const { in get_output_index() function in executorch::runtime::Method
1362 return get_value(get_output_index(i)); in get_output()
1366 return mutable_value(get_output_index(i)); in mutable_output()
Dmethod.h311 size_t get_output_index(size_t i) const;
/external/tensorflow/tensorflow/core/framework/
Dop_kernel.cc502 TF_RETURN_IF_ERROR(get_output_index(output_name, &output_index)); in forward_input_to_output_with_shape()
884 Status OpKernelContext::get_output_index(StringPiece name, in get_output_index() function in tensorflow::OpKernelContext
900 TF_RETURN_IF_ERROR(get_output_index(name, &index)); in set_output()
907 TF_RETURN_IF_ERROR(get_output_index(name, &index)); in set_output()
1015 TF_RETURN_IF_ERROR(get_output_index(name, &index)); in set_output_ref()
1022 TF_RETURN_IF_ERROR(get_output_index(name, &index)); in mutable_output()
Dop_kernel.h1256 Status get_output_index(StringPiece name, int* out_index) const;