Searched defs:GatherArguments (Results 1 – 1 of 1) sorted by relevance
410 struct GatherArguments { struct in tensorflow::__anond7c25eea0111::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;