Searched defs:param_axis (Results 1 – 1 of 1) sorted by relevance
30 param_axis: str = "axis" variable in OpConcat124 param_axis: str = "axis" variable in OpExpandDims136 param_axis: str = "axis" variable in OpGather171 param_axis: str = "axis" variable in OpLogSoftmax185 param_axis: str = "axis" variable in OpPack308 param_axis: str = "axis" variable in OpSoftmax326 param_axis: str = "axis" variable in OpSplit