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