Searched defs:data_inputs (Results 1 – 9 of 9) sorted by relevance
77 OpInputList* data_inputs, int* first_dim_size, in CheckArgsAndAllocateResult()165 OpInputList data_inputs; in Compute() local237 OpInputList data_inputs; in Compute() local
42 std::vector<DatasetBase*> data_inputs) in Dataset()299 std::vector<DatasetBase*> data_inputs; in MakeDataset() local
107 def __init__(self, selector_input, data_inputs): argument
243 std::vector<NodeBuilder::NodeOut> data_inputs; in RewriteIntoAddN() local
270 std::vector<std::pair<Node*, int>> data_inputs(num_inputs); in BuildXlaLaunchOps() local
1172 int data_inputs = 0; in DetermineCondStateMerge() local
1123 std::set<const Node*> data_inputs; in AddStrictInputSemantics() local
476 std::vector<std::pair<int, int>> data_inputs; member
2700 std::vector<std::pair<Node*, int>> data_inputs(num_inputs); in BuildTPUReplicateOps() local