Searched refs:axis_data (Results 1 – 4 of 4) sorted by relevance
46 const auto& axis_data = indices_array.GetBuffer<ArrayDataType::kInt32>().data; in Run() local47 CHECK_EQ(axis_data.size(), 1) in Run()49 op->axis = {axis_data[0]}; in Run()
64 const auto& axis_data = axis_array.GetBuffer<ArrayDataType::kInt32>().data; in Run() local65 CHECK_EQ(axis_data.size(), 1); in Run()66 const int axis = axis_data[0]; in Run()
182 def as_axis(axis_data): argument191 if isinstance(axis_data, Axis):192 axis = axis_data194 axis = Axis(*axis_data)217 for axis_data in axes:218 axis = as_axis(axis_data)
258 void ResolveAxis(const int* axis_data, int axis_count, in ResolveAxis() argument262 op_params->axis[i] = static_cast<int16>(axis_data[i]); in ResolveAxis()