Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Dremote_tensor_handle_data.h44 int32 output_num() const { return output_num_; } in output_num() function
77 int32 output_num() const { return output_num_; } in output_num() function
Dremote_tensor_handle_data.cc28 uint64 context_id, uint64 op_id, int output_num, in DestroyRemoteTensorHandle()
86 RemoteTensorHandleData::RemoteTensorHandleData(int64 op_id, int output_num, in RemoteTensorHandleData()
151 int64 op_id, int32 output_num, const string& remote_task, uint64 context_id, in UnshapedRemoteTensorHandleData()
Dremote_mgr_test.cc79 const int output_num = 3; in TEST_F() local
98 const int output_num = 1; in TEST_F() local
Dremote_tensor_handle.h31 int32 output_num; member
Dremote_mgr.cc77 int64* op_id, int32* output_num) { in GetRemoteTensorHandle()
119 int32 output_num; in SerializeRemoteTensorHandle() local
/external/tensorflow/tensorflow/compiler/jit/
Dxla_launch_util.cc252 const xla::HloInputOutputAliasConfig& input_output_alias, int output_num) { in MustAliasOutput()
271 int output_num, OpKernelContext* ctx, int missing_ctx_input_prefix, in FindAliasedTensorForOutput()
307 int output_num, OpKernelContext* ctx, int missing_ctx_input_prefix, in GetOrCreateTensorForOutput()
323 const xla::HloInputOutputAliasConfig& input_output_alias, int output_num, in SetBufferForTensorUnderAllocateXlaTensors()
349 const xla::HloInputOutputAliasConfig& input_output_alias, int output_num, in SetBufferForResourceVarTensorUnderAllocateXlaTensors()
416 int output_num = 0; in PopulateOutputs() local
/external/tensorflow/tensorflow/core/protobuf/
Dremote_tensor_handle.proto24 int32 output_num = 2; field
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dpromote_resources_to_args.cc142 int64_t output_num = return_op.getNumOperands(); in PromoteResourcesToArguments() local
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dtensor_handle.cc202 Status TensorHandle::CreateRemoteHandle(int64 op_id, int output_num, in CreateRemoteHandle()
242 int64 op_id, int32 output_num, const string& remote_task, uint64 context_id, in CreateUnshapedRemoteHandle()
448 const int64 op_id, const int32 output_num) { in SetRemoteOpIdAndOutputNumToLocalTensorHandle()
/external/grpc-grpc/test/core/util/
Dtest_config.cc228 static void output_num(long num) { in output_num() function
/external/tensorflow/tensorflow/compiler/tf2xla/
Dgraph_compiler_util.cc251 int64 output_num = config.fetch_size(); in PopulateXlaArgsAndXlaAlias() local
/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.cc561 for (auto output_num : op.outputs) { in ConvertOp() local