Searched refs:tensor_info_dxdy (Results 1 – 2 of 2) sorted by relevance
79 const TensorInfo tensor_info_dxdy(shape, Format::F32); in configure() local82 _impl->dx.allocator()->init(tensor_info_dxdy); in configure()83 _impl->dy.allocator()->init(tensor_info_dxdy); in configure()
117 TensorInfo tensor_info_dxdy(shape, Format::F32); in configure() local119 auto dx = std::make_unique<TensorInfo>(tensor_info_dxdy); in configure()120 auto dy = std::make_unique<TensorInfo>(tensor_info_dxdy); in configure()