Home
last modified time | relevance | path

Searched defs:output_num (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Dremote_mgr_test.cc75 const int output_num = 3; in TEST_F() local
94 const int output_num = 1; in TEST_F() local
114 const int output_num = 3; in TEST_F() local
135 const int output_num = 1; in TEST_F() local
Dremote_tensor_handle_data.cc29 uint64 context_id, uint64 op_id, int output_num, in DestroyRemoteTensorHandle()
87 RemoteTensorHandleData::RemoteTensorHandleData(int64 op_id, int output_num, in RemoteTensorHandleData()
100 RemoteTensorHandleData::RemoteTensorHandleData(int64 op_id, int output_num, in RemoteTensorHandleData()
Dremote_mgr.cc39 int64 operation_id, int32 output_num) { in AddOperationOutput()
85 int64* op_id, int32* output_num) { in GetRemoteTensorHandle()
128 int32 output_num; in SerializeRemoteTensorHandle() local
Dremote_tensor_handle.h31 int32 output_num; member
Deager_service_impl.cc195 const int output_num = output_nums.empty() ? i : output_nums.at(i); in AddOpRetvalsToResponse() local
475 for (const int32 output_num : request->output_num()) { in RunComponentFunction() local
Deager_service_impl_test.cc227 const int output_num, RunComponentFunctionRequest* request) { in AddOperationToRunComponentFunctionRequest()
614 const int output_num = 5; in TestComponentFunction() local
1197 int32 output_num; in TEST_F() local
/external/tensorflow/tensorflow/core/protobuf/
Dremote_tensor_handle.proto23 int32 output_num = 2; field
Deager_service.proto185 repeated int32 output_num = 3; field
/external/tensorflow/tensorflow/compiler/jit/
Dxla_launch_util.cc302 int output_num, OpKernelContext* ctx, int missing_ctx_input_prefix, in GetOrCreateTensorForOutput()
331 xla::ScopedShapedBuffer* output, int output_num, in PopulateXlaTensor()
345 const XlaCompiler::CompilationResult* compilation_result, int output_num) { in SetOutputForConstant()
487 int output_num = 0; in PopulateOutputs() local
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dtensor_handle.cc369 int64 op_id, int32 output_num, const string& remote_task, in CreateUnshapedRemoteHandle()
376 TensorHandle::TensorHandle(int64 op_id, int32 output_num, in TensorHandle()
396 int64 op_id, int32 output_num, tensorflow::DataType dtype, Device* d, in CreateLazyRemoteHandle()
401 TensorHandle::TensorHandle(int64 op_id, int32 output_num, in TensorHandle()
770 int output_num, in AddUnshapedRemoteMirror()
795 int output_num, EagerContext* ctx) { in AddResourceShapeMirror()
Dexecute.cc575 const KernelAndDevice& kernel, const int output_num, Device* output_device, in CreateUnshapedOutput()
/external/grpc-grpc/test/core/util/
Dtest_config.cc228 static void output_num(long num) { in output_num() function
/external/libaom/libaom/av1/encoder/
Dcnn.h134 int output_num; // The output buffer idx to which the layer output is member
Dcnn.c279 const int output_num = layer_config->output_num; in av1_find_cnn_output_size() local
876 const int output_num = layer_config->output_num; in av1_cnn_predict_c() local
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dflatbuffer_import.cc659 for (auto output_num : op.outputs) { in ConvertOp() local
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.cc3281 int output_num = in BuildExecuteNodes() local