| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| H A D | remote_mgr_test.cc | 78 const int output_num = 3; in TEST_F() local
97 const int output_num = 1; in TEST_F() local
117 const int output_num = 3; in TEST_F() local
138 const int output_num = 1; in TEST_F() local
161 const int output_num = 1; in TEST_F() local
|
| H A D | remote_tensor_handle_data.cc | 30 uint64 context_id, uint64 op_id, int output_num, in DestroyRemoteTensorHandle()
88 RemoteTensorHandleData::RemoteTensorHandleData(int64_t op_id, int output_num, in RemoteTensorHandleData()
101 RemoteTensorHandleData::RemoteTensorHandleData(int64_t op_id, int output_num, in RemoteTensorHandleData()
|
| H A D | remote_mgr.cc | 55 int64_t operation_id, int32_t output_num) { in AddOperationOutput()
113 int64_t* op_id, int32* output_num) { in GetRemoteTensorHandle()
156 int32_t output_num; in SerializeRemoteTensorHandle() local
|
| H A D | eager_service_impl.cc | 200 const int output_num = output_nums.empty() ? i : output_nums.at(i); in AddOpRetvalsToResponse() local
515 for (const int32_t output_num : request->output_num()) { in RunComponentFunction() local
|
| H A D | remote_tensor_handle.h | 31 int32 output_num; member
|
| H A D | eager_service_impl_test.cc | 233 const int output_num, RunComponentFunctionRequest* request) { in AddOperationToRunComponentFunctionRequest()
620 const int output_num = 5; in TestComponentFunction() local
1215 int32_t output_num; in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/ |
| H A D | tensor_handle.cc | 369 int64_t op_id, int32_t output_num, const string& remote_task, in CreateUnshapedRemoteHandle()
376 TensorHandle::TensorHandle(int64_t op_id, int32_t output_num, in TensorHandle()
396 int64_t op_id, int32_t output_num, tensorflow::DataType dtype, Device* d, in CreateLazyRemoteHandle()
401 TensorHandle::TensorHandle(int64_t op_id, int32_t output_num, in TensorHandle()
770 int output_num, in AddUnshapedRemoteMirror()
795 int output_num, EagerContext* ctx) { in AddResourceShapeMirror()
|
| H A D | execute.cc | 1274 const KernelAndDevice& kernel, const int output_num, Device* output_device, in CreateUnshapedOutput()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A D | xla_launch_util.cc | 321 xla::ScopedShapedBuffer& output, int output_num, OpKernelContext* ctx, in GetOrCreateTensorForOutput()
380 const XlaCompiler::CompilationResult* compilation_result, int output_num) { in SetOutputForConstant()
539 int output_num = 0; in PopulateOutputs() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/ |
| H A D | remote_tensor_handle.proto | 23 int32 output_num = 2; field
|
| H A D | eager_service.proto | 185 repeated int32 output_num = 3; field
|
| /aosp_15_r20/external/libaom/av1/encoder/ |
| H A D | cnn.h | 135 int output_num; // The output buffer idx to which the layer output is member
|
| H A D | cnn.c | 283 const int output_num = layer_config->output_num; in av1_find_cnn_output_size() local
945 const int output_num = layer_config->output_num; in av1_cnn_predict_c() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ir/importexport/ |
| H A D | functiondef_import.cc | 93 int output_num = 0; in GetValueOrCreatePlaceholder() local
|
| /aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/core/protobuf/ |
| D | remote_tensor_handle.pb.h | 568 inline ::int32_t RemoteTensorHandle::output_num() const { in output_num() function
|
| D | eager_service.pb.h | 6366 inline ::int32_t RunComponentFunctionRequest::output_num(int index) const { in output_num() function
6386 RunComponentFunctionRequest::output_num() const { in output_num() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | light_outside_compilation.cc | 276 int GetOutputBufferId(int output_num, const TfCallbackData& callback_data) { in GetOutputBufferId()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/ |
| H A D | flatbuffer_import.cc | 818 for (auto output_num : op.outputs) { in ConvertOp() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| H A D | distributed_tpu_rewrite_pass.cc | 3718 int output_num = in BuildExecuteNodes() local
|