Searched refs:FunctionArgIndex (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | partitioning_utils.cc | 78 std::vector<FunctionArgIndex>* arg_indices, std::vector<int>* ret_indices, in UpdateArgAndRetvalMetadata()
81 std::vector<std::pair<Node*, FunctionArgIndex>> arg_nodes; in UpdateArgAndRetvalMetadata()
95 arg_nodes.emplace_back(node, FunctionArgIndex(index, sub_index)); in UpdateArgAndRetvalMetadata()
107 auto arg_comparator = [](std::pair<Node*, FunctionArgIndex> a, in UpdateArgAndRetvalMetadata()
108 std::pair<Node*, FunctionArgIndex> b) { in UpdateArgAndRetvalMetadata()
|
| D | partitioning_utils_test.cc | 169 void CheckArgIndices(const std::vector<FunctionArgIndex>& expected, in CheckArgIndices()
170 const std::vector<FunctionArgIndex>& actual) { in CheckArgIndices()
197 std::vector<FunctionArgIndex> arg_indices; in TEST_F()
236 std::vector<FunctionArgIndex> arg_indices; in TEST_F()
|
| D | partitioning_utils.h | 65 std::vector<FunctionArgIndex>* arg_indices, std::vector<int>* ret_indices,
|
| D | process_function_library_runtime.h | 46 virtual Status GetLocalArg(const FunctionArgIndex& index,
52 virtual Status GetRemoteArg(const FunctionArgIndex& index, in GetRemoteArg()
246 std::vector<FunctionArgIndex> arg_indices;
|
| D | process_function_library_runtime_test.cc | 825 Status GetLocalArg(const FunctionArgIndex& index, in GetLocalArg()
|
| D | process_function_library_runtime.cc | 1682 const FunctionArgIndex index = comp_data.arg_indices.at(i); in Run()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | execute_node_test.cc | 117 TF_EXPECT_OK(args.GetLocalArg(FunctionArgIndex(0), &local0)); in TEST()
121 TF_EXPECT_OK(args.GetLocalArg(FunctionArgIndex(1, 0), &local1)); in TEST()
125 TF_EXPECT_OK(args.GetRemoteArg(FunctionArgIndex(1, 1), &remote0)); in TEST()
129 TF_EXPECT_OK(args.GetRemoteArg(FunctionArgIndex(2), &remote1)); in TEST()
|
| D | execute_node.h | 58 Status GetLocalArg(const FunctionArgIndex& index, Tensor* val) const override;
65 Status GetRemoteArg(const FunctionArgIndex& index, in GetRemoteArg()
87 std::function<Status(const FunctionArgIndex&, eager::RemoteTensorHandle*)>
|
| D | execute_node.cc | 102 const FunctionArgIndex& index, in Init()
124 Status ExecuteNodeArgs::GetLocalArg(const FunctionArgIndex& index, in GetLocalArg()
|
| D | kernel_and_device.h | 76 Status GetLocalArg(const FunctionArgIndex& index, Tensor* val) const override;
|
| D | kernel_and_device.cc | 53 Status EagerKernelArgs::GetLocalArg(const FunctionArgIndex& index, in GetLocalArg()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function.h | 599 struct FunctionArgIndex { struct
600 explicit FunctionArgIndex(const int index) : index(index) {} in FunctionArgIndex() function
601 FunctionArgIndex(const int index, const int sub_index) in FunctionArgIndex() function
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl_test.cc | 722 Status GetRemoteArg(const FunctionArgIndex& index, in GetRemoteArg()
|