Searched refs:depth_val (Results 1 – 1 of 1) sorted by relevance
159 int64_t depth_val = ShapedType::kDynamicSize; in InferOneHotOpType() local163 depth_val = (*depth_attr.begin()).getSExtValue(); in InferOneHotOpType()164 shape.insert(shape.begin() + axis_val, depth_val); in InferOneHotOpType()