Searched defs:GatherArguments (Results 1 – 1 of 1) sorted by relevance
410 struct GatherArguments { struct in tensorflow::__anon92bff46a0111::OpTest411 int64_t batch_dims;412 DataType axis_type;413 DataType indices_type;414 DataType params_type;415 std::vector<int64_t> params_shape;416 Tensor indices;417 Tensor axis;