Searched refs:depth_array (Results 1 – 1 of 1) sorted by relevance
1968 const auto& depth_array = in ProcessOneHotOperator() local1970 if (!depth_array.has_shape()) { in ProcessOneHotOperator()1974 CHECK(depth_array.data_type == ArrayDataType::kInt32) in ProcessOneHotOperator()1976 CHECK_EQ(RequiredBufferSizeForShape(depth_array.shape()), 1) in ProcessOneHotOperator()1979 const int depth = depth_array.GetBuffer<ArrayDataType::kInt32>().data[0]; in ProcessOneHotOperator()