Searched refs:FunctionArgIndex (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | execute_node_test.cc | 122 TF_EXPECT_OK(args.GetLocalArg(FunctionArgIndex(0), &local0)); in TEST()
126 TF_EXPECT_OK(args.GetLocalArg(FunctionArgIndex(1, 0), &local1)); in TEST()
130 TF_EXPECT_OK(args.GetRemoteArg(FunctionArgIndex(1, 1), &remote0)); in TEST()
134 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 | 82 Status GetLocalArg(const FunctionArgIndex& index, Tensor* val) const override;
|
| D | kernel_and_device.cc | 54 Status EagerKernelArgs::GetLocalArg(const FunctionArgIndex& index, in GetLocalArg()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | partitioning_utils.cc | 155 Graph* graph, std::vector<FunctionArgIndex>* arg_indices, in UpdateArgAndRetvalMetadata()
159 std::vector<std::pair<Node*, FunctionArgIndex>> arg_nodes; in UpdateArgAndRetvalMetadata()
173 arg_nodes.emplace_back(node, FunctionArgIndex(index, sub_index)); in UpdateArgAndRetvalMetadata()
185 auto arg_comparator = [](std::pair<Node*, FunctionArgIndex> a, in UpdateArgAndRetvalMetadata()
186 std::pair<Node*, FunctionArgIndex> b) { in UpdateArgAndRetvalMetadata()
|
| D | partitioning_utils_test.cc | 260 void CheckArgIndices(const std::vector<FunctionArgIndex>& expected, in CheckArgIndices()
261 const std::vector<FunctionArgIndex>& actual) { in CheckArgIndices()
288 std::vector<FunctionArgIndex> arg_indices; in TEST_F()
314 std::vector<FunctionArgIndex> arg_indices; in TEST_F()
332 std::vector<FunctionArgIndex> arg_indices; in TEST_F()
361 std::vector<FunctionArgIndex> arg_indices; in TEST_F()
|
| D | partitioning_utils.h | 77 Graph* graph, std::vector<FunctionArgIndex>* arg_indices,
|
| D | process_function_library_runtime.h | 47 virtual Status GetLocalArg(const FunctionArgIndex& index,
53 virtual Status GetRemoteArg(const FunctionArgIndex& index, in GetRemoteArg()
279 std::vector<FunctionArgIndex> arg_indices;
|
| D | process_function_library_runtime_test.cc | 874 Status GetLocalArg(const FunctionArgIndex& index, in GetLocalArg()
|
| D | process_function_library_runtime.cc | 1701 const FunctionArgIndex index = comp_data.arg_indices.at(i); in GetComponentArgs()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function.h | 621 struct FunctionArgIndex { struct
622 explicit FunctionArgIndex(const int index) : index(index) {} in FunctionArgIndex() function
623 FunctionArgIndex(const int index, const int sub_index) in FunctionArgIndex() function
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl_test.cc | 728 Status GetRemoteArg(const FunctionArgIndex& index, in GetRemoteArg()
|